中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

Scikit-learn中怎么實現模型持久化

小億
83
2024-05-10 18:20:57
欄目: 編程語言

在Scikit-learn中,可以使用joblib庫來實現模型的持久化。joblib庫提供了dump和load函數,可以方便地將模型保存到文件或者從文件中加載模型。

以下是一個示例代碼,演示了如何使用joblib庫來保存和加載一個訓練好的模型:

from sklearn import datasets
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
from sklearn import metrics
from sklearn.externals import joblib

# 加載數據集
iris = datasets.load_iris()
X = iris.data
y = iris.target

# 劃分訓練集和測試集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)

# 訓練模型
model = RandomForestClassifier()
model.fit(X_train, y_train)

# 保存模型
joblib.dump(model, 'random_forest_model.pkl')

# 加載模型
loaded_model = joblib.load('random_forest_model.pkl')

# 使用加載的模型進行預測
y_pred = loaded_model.predict(X_test)

# 評估模型
print("Accuracy:", metrics.accuracy_score(y_test, y_pred))

在上面的代碼中,我們首先訓練了一個隨機森林分類器模型,然后使用joblib庫的dump函數將模型保存到文件’random_forest_model.pkl’中。接著使用load函數加載模型,并使用加載的模型進行預測和評估。

0
正蓝旗| 商水县| 濮阳市| 平谷区| 京山县| 苏尼特右旗| 二连浩特市| 桦甸市| 莲花县| 阿拉善盟| 尼木县| 垫江县| 新泰市| 盐池县| 廉江市| 南城县| 庆城县| 灵川县| 尼勒克县| 南平市| 大埔县| 绵阳市| 伊宁县| 元氏县| 西青区| 鄂尔多斯市| 洛阳市| 英德市| 沅江市| 邵阳市| 佛山市| 晋江市| 内丘县| 双流县| 崇州市| 梅河口市| 九龙坡区| 敦化市| 贵定县| 华亭县| 沛县|