Bachstelze commited on
Commit
99097e8
·
1 Parent(s): 29323a5

update pickle with classes

Browse files
A5b/cv_baseline.py CHANGED
@@ -203,6 +203,7 @@ artifact = {
203
  'model_name' : best_name,
204
  'scaler' : scaler,
205
  'feature_columns' : feature_columns,
 
206
  'cv_metrics': {
207
  'f1_mean' : float(results_df.iloc[0]['F1_mean']),
208
  'f1_std' : float(results_df.iloc[0]['F1_std']),
 
203
  'model_name' : best_name,
204
  'scaler' : scaler,
205
  'feature_columns' : feature_columns,
206
+ 'classes': list(best_model.classes_),
207
  'cv_metrics': {
208
  'f1_mean' : float(results_df.iloc[0]['F1_mean']),
209
  'f1_std' : float(results_df.iloc[0]['F1_std']),
A5b/models/ensemble_classification_champion.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bca3b9c29ceaa2623e96f782f2a15625d54fa044acfe128673b60f29f2b56025
3
- size 30798304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c10163833b9634e6f75ad58285deec7d134ac8ec41802e80c31bde1523208a20
3
+ size 30798356