Scikit-learn中有幾種常見的部署模型的方法:
import pickle
# 訓練模型
# model.fit(X_train, y_train)
# 保存模型
with open('model.pkl', 'wb') as f:
pickle.dump(model, f)
# 加載模型
with open('model.pkl', 'rb') as f:
model = pickle.load(f)
# 在部署時使用模型進行預測
# y_pred = model.predict(X_test)
from joblib import dump, load
# 訓練模型
# model.fit(X_train, y_train)
# 保存模型
dump(model, 'model.joblib')
# 加載模型
model = load('model.joblib')
# 在部署時使用模型進行預測
# y_pred = model.predict(X_test)
from flask import Flask, request, jsonify
app = Flask(__name__)
@app.route('/predict', methods=['POST'])
def predict():
data = request.get_json()
X = data['X']
y_pred = model.predict(X)
return jsonify({'y_pred': y_pred.tolist()})
if __name__ == '__main__':
app.run()
以上是幾種常見的Scikit-learn部署模型的方法,選擇適合自己場景的方式進行部署。