Skip to content

Commit cd22a2c

Browse files
Train.py Local Path Issue Attempt #3
1 parent a6f5971 commit cd22a2c

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

train.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,10 +49,8 @@ def load_and_prepare_data():
4949
def train_model(X_train, X_test, y_train, y_test, n_estimators=100, max_depth=5, random_state=42):
5050
"""Обучение модели Random Forest с логированием в MLflow"""
5151

52-
# Настройка MLflow для работы в локальной директории
53-
mlflow.set_tracking_uri("file:./mlruns")
54-
5552
# Установка имени эксперимента
53+
# (tracking URI уже настроен через переменные окружения в начале файла)
5654
mlflow.set_experiment("iris_classification")
5755

5856
with mlflow.start_run():

0 commit comments

Comments
 (0)