НачатьНачать бесплатно

Сохранение и загрузка модели

С помощью Model API разработчики могут обмениваться моделями, даже если у них нет доступа к одному серверу MLflow Tracking, — для этого используется локальная файловая система.

В этом упражнении вы обучите новую модель LinearRegression на основе существующей, используя набор данных Unicorn. Сначала вы загрузите существующую модель из локальной файловой системы, затем обучите на её основе новую модель и сохраните её обратно в локальную файловую систему.

Существующая модель сохранена в локальной файловой системе в директории "lr_local_v1". Модуль mlflow уже импортирован.

Это упражнение является частью курса

Введение в MLflow

Посмотреть курс

Инструкции к упражнению

  • Загрузите модель из директории "lr_local_v1" локальной файловой системы, используя библиотеку scikit-learn из модуля MLflow.
  • С помощью библиотеки scikit-learn из модуля mlflow сохраните модель локально в директорию "lr_local_v2".

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Load model from local filesystem
model = ____.____.____("____")

# Training Data
X = df[["R&D Spend", "Administration", "Marketing Spend", "State"]]
y = df[["Profit"]]
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=0.7,random_state=0)
# Train Model
model.fit(X_train, y_train)

# Save model to local filesystem
____.____.____(____, "____")
Редактировать и запускать код