Skip to content

Commit 97a0b6e

Browse files
Added Experiment Folders Handling
1 parent 8cff1ed commit 97a0b6e

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

train.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,8 +90,12 @@ def train_model(X_train, X_test, y_train, y_test, n_estimators=100, max_depth=5,
9090
print(f"Текущая рабочая директория: {os.getcwd()}")
9191
print(f"HOME: {os.environ.get('HOME', 'не установлен')}")
9292

93-
# Установка имени эксперимента
94-
experiment_name = "iris_classification"
93+
# Каждый запуск создает новый эксперимент с уникальным именем
94+
# Используем timestamp для уникальности
95+
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
96+
experiment_name = f"iris_classification_{timestamp}"
97+
98+
print(f"Создание нового эксперимента: {experiment_name}")
9599
mlflow.set_experiment(experiment_name)
96100

97101
with mlflow.start_run():

0 commit comments

Comments
 (0)