Spaces:
Running
Running
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:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c10163833b9634e6f75ad58285deec7d134ac8ec41802e80c31bde1523208a20
|
| 3 |
+
size 30798356
|