Как сохранить ансамбль моделей машинного обучения на моем P C?

Question

Я использую Google Colab, и время выполнения отключается. Вместо того, чтобы тренировать это снова и снова, я думал, что каким-то образом сохраню это в P C. Так что, если мне придется продемонстрировать это, я могу просто загрузить этот файл и использовать .predict. Мне нужно сохранить его на моем P C, чтобы я мог запустить его с помощью простой функции .load (), если что-то отключается на платформе, которую я использую.

Answer 1

Если вы используете python, вы можете использовать травление, чтобы сохранить как объект

import pickle
model = LogisticRegression()
model.fit(X_train, Y_train)
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))
# load the model from disk
loaded_model = pickle.load(open(filename, 'rb'))
result = loaded_model.score(X_test, Y_test)
print(result)