Asif Ahmad
commited on
Commit
·
c7a9443
1
Parent(s):
059311a
Update xgb-mining-model.py
Browse files- xgb-mining-model.py +1 -1
xgb-mining-model.py
CHANGED
@@ -133,7 +133,7 @@ class BaseMiningModel:
|
|
133 |
# restore_best_weights=True)
|
134 |
|
135 |
model.fit(X_train, Y_train)#, epochs=epochs, batch_size=self.batch_size, callbacks=[early_stopping])
|
136 |
-
model.
|
137 |
|
138 |
def predict(self, data: ndarray):
|
139 |
predictions = []
|
|
|
133 |
# restore_best_weights=True)
|
134 |
|
135 |
model.fit(X_train, Y_train)#, epochs=epochs, batch_size=self.batch_size, callbacks=[early_stopping])
|
136 |
+
model.save_model(self.model_dir)
|
137 |
|
138 |
def predict(self, data: ndarray):
|
139 |
predictions = []
|