Сохранение и загрузка модели
С помощью 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
____.____.____(____, "____")