from sklearn.linear_model import LinearRegression from sklearn.model_selection import train_test_split import numpy as np import pandas as pd
# Load the data data = pd.read_csv('data.csv')
# Split the data into training and testing sets X_train, X_test, y_train, y_test = train_test_split(data[['input_feature']], data['output'], test_size=0.2)
# Train the model model = LinearRegression() model.fit(X_train, y_train)
# Evaluate the model on the testing data print(model.score(X_test, y_test))
Flask 是一个轻量级的 Web 框架,可用于创建简单的 Web 应用程序。要使用 Flask 部署机器学习模型,您可以创建一个 Flask 应用程序来接收用户的输入、使用模型进行预测并返回结果。
以下是如何使用 Flask 部署机器学习模型的示例:
from flask import Flask, request, jsonify import pickle
app = Flask(__name__)
# Load the saved model model = pickle.load(open("model.pkl", "rb"))
# Define a route for making predictions @app.route("/predict", methods=["POST"]) def predict(): # Get the input data data = request.get_json(force=True) prediction = model.predict([[data["input_1"], data["input_2"], data["input_3"]]])
# Return the prediction return jsonify(prediction[0])
if __name__ == "__main__": app.run()
复制
FastAPI 是一个更现代的 Web 框架,旨在快速且易于使用。它可用于创建能够处理大量请求并快速返回结果的 Web 应用程序。要使用 FastAPI 部署机器学习模型,您可以创建一个 FastAPI 应用程序来接收用户的输入、使用模型进行预测并返回结果。
以下是如何使用 FastAPI 部署机器学习模型的示例:
from fastapi import FastAPI import pickle
app = FastAPI()
# Load the saved model model = pickle.load(open("model.pkl", "rb"))
# Define a route for making predictions @app.post("/predict") def predict(input_1: float, input_2: float, input_3: float): prediction = model.predict([[input_1, input_2, input_3]]) return {"prediction": prediction[0]}