Spaces:
Sleeping
Sleeping
fix syntax error
Browse files- src/model_utils.py +2 -2
src/model_utils.py
CHANGED
|
@@ -70,7 +70,7 @@ def save_model(model, user_model_name, metrics_result_single=None):
|
|
| 70 |
login(token=os.environ["HF_TOKEN"])
|
| 71 |
|
| 72 |
timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
|
| 73 |
-
filename = f"{timestamp}{st.session_state.get(
|
| 74 |
|
| 75 |
# Prepare model dict (same as before)
|
| 76 |
model_data = {
|
|
@@ -142,7 +142,7 @@ def save_model_ensemble(models, user_model_name, best_iterations=None, fold_scor
|
|
| 142 |
login(token=os.environ["HF_TOKEN"])
|
| 143 |
|
| 144 |
timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
|
| 145 |
-
filename = f"{timestamp}{st.session_state.get(
|
| 146 |
|
| 147 |
ensemble_data = {
|
| 148 |
"timestamp": timestamp,
|
|
|
|
| 70 |
login(token=os.environ["HF_TOKEN"])
|
| 71 |
|
| 72 |
timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
|
| 73 |
+
filename = f"{timestamp}{st.session_state.get('target_col', 'UNKNOWN')[0]}_{user_model_name}_single"
|
| 74 |
|
| 75 |
# Prepare model dict (same as before)
|
| 76 |
model_data = {
|
|
|
|
| 142 |
login(token=os.environ["HF_TOKEN"])
|
| 143 |
|
| 144 |
timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
|
| 145 |
+
filename = f"{timestamp}{st.session_state.get('target_col', 'UNKNOWN')[0]}_{user_model_name}_ensemble"
|
| 146 |
|
| 147 |
ensemble_data = {
|
| 148 |
"timestamp": timestamp,
|