ensure predict_proba follows suit. Remove all lib specific params from example config

This commit is contained in:
robcaulk 2022-12-11 12:39:31 +01:00
parent 85f22b5c30
commit 8c7ec07951
2 changed files with 2 additions and 1 deletions

View File

@ -80,7 +80,6 @@
"random_state": 1
},
"model_training_parameters": {
"n_estimators": 1000
}
},
"bot_name": "",

View File

@ -101,6 +101,8 @@ class BaseClassifierModel(IFreqaiModel):
pred_df = DataFrame(predictions, columns=dk.label_list)
predictions_prob = self.model.predict_proba(dk.data_dictionary["prediction_features"])
if self.CONV_WIDTH == 1:
predictions_prob = np.reshape(predictions_prob, (-1, len(self.model.classes_)))
pred_df_prob = DataFrame(predictions_prob, columns=self.model.classes_)
pred_df = pd.concat([pred_df, pred_df_prob], axis=1)