Asif Ahmad commited on
Commit
c7a9443
·
1 Parent(s): 059311a

Update xgb-mining-model.py

Browse files
Files changed (1) hide show
  1. 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.save(self.model_dir)
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 = []