Skip to content

Commit 3be4839

Browse files
committed
Add sklearn and joblib versions to downloaded model file names
Remove debug print statement
1 parent d4b84c2 commit 3be4839

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

cesium_app/handlers/feature.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,9 @@ def get(self, featureset_id=None, action=None):
2323
featureset = Featureset.get_if_owned_by(featureset_id,
2424
self.current_user)
2525
fset_path = featureset.file_uri
26-
print(fset_path)
2726
fset, data = featurize.load_featureset(fset_path)
2827
fset.index.name = 'ts_name'
29-
fset.columns = fset.columns.get_level_values(0)
28+
fset.columns = fset.columns.droplevel('channel')
3029
fset.columns.name = None
3130
self.set_header("Content-Type", 'text/csv; charset="utf-8"')
3231
self.set_header(

cesium_app/handlers/model.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import uuid
1414
import datetime
1515

16+
import sklearn
1617
from sklearn.model_selection import GridSearchCV
1718
import joblib
1819

@@ -81,8 +82,10 @@ def get(self, model_id=None, action=None):
8182
self.set_header("Content-Type", "application/octet-stream")
8283
self.set_header(
8384
"Content-Disposition", "attachment; "
84-
f"filename=cesium_model__joblib_{model.project.name}"
85-
f"_{model.name}_{model.finished}.pkl")
85+
f"filename=cesium_model_{model.project.name}"
86+
f"_{model.name}_{str(model.finished).replace(' ', 'T')}"
87+
f"_joblib_v{joblib.__version__}"
88+
f"_sklearn_v{sklearn.__version__}.pkl")
8689
self.write(model_data)
8790
else:
8891
if model_id is not None:

0 commit comments

Comments
 (0)