如何在 python 中保存和重用模型
How to save and reuse model in python
我已经构建了一个模型并使用 pickle 库保存了它,但我想通过更改输入变量的值再次使用它,为此我没有任何输出。
这是代码:
# Save Model Using Pickle
import pandas
from sklearn import model_selection
from sklearn.linear_model import LogisticRegression
import pickle
url = "https://raw.githubusercontent.com/jbrownlee/Datasets/master/pima-indians-
diabetes.data.csv"
names = ['preg', 'plas', 'pres', 'skin', 'test', 'mass', 'pedi', 'age', 'class']
dataframe = pandas.read_csv(url, names=names)
array = dataframe.values
X = array[:,0:8]
Y = array[:,8]
test_size = 0.33
seed = 7
X_train, X_test, Y_train, Y_test = model_selection.train_test_split(X, Y, test_size=test_size,
random_state=seed)
# Fit the model on training set
model = LogisticRegression()
model.fit(X_train, Y_train)
# save the model to disk
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))
# some time later...
# load the model from disk
loaded_model = pickle.load(open(filename, 'rb'))
result = loaded_model.score(X_test, Y_test)
print(result)
现在,我只有 X_test 的数据。我的 Y_test 没有任何数据点。在这种情况下,我如何再次使用它并仅传递 X_test 它给了我 Y。
请帮忙!!
但您可以使用像
这样的简单命令
l.to_pickle('filename.pkl')
所以,你可以轻松使用它!
这应该有效:
loaded_model.predict(X_test)
我已经构建了一个模型并使用 pickle 库保存了它,但我想通过更改输入变量的值再次使用它,为此我没有任何输出。
这是代码:
# Save Model Using Pickle
import pandas
from sklearn import model_selection
from sklearn.linear_model import LogisticRegression
import pickle
url = "https://raw.githubusercontent.com/jbrownlee/Datasets/master/pima-indians-
diabetes.data.csv"
names = ['preg', 'plas', 'pres', 'skin', 'test', 'mass', 'pedi', 'age', 'class']
dataframe = pandas.read_csv(url, names=names)
array = dataframe.values
X = array[:,0:8]
Y = array[:,8]
test_size = 0.33
seed = 7
X_train, X_test, Y_train, Y_test = model_selection.train_test_split(X, Y, test_size=test_size,
random_state=seed)
# Fit the model on training set
model = LogisticRegression()
model.fit(X_train, Y_train)
# save the model to disk
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))
# some time later...
# load the model from disk
loaded_model = pickle.load(open(filename, 'rb'))
result = loaded_model.score(X_test, Y_test)
print(result)
现在,我只有 X_test 的数据。我的 Y_test 没有任何数据点。在这种情况下,我如何再次使用它并仅传递 X_test 它给了我 Y。 请帮忙!!
但您可以使用像
这样的简单命令l.to_pickle('filename.pkl')
所以,你可以轻松使用它!
这应该有效:
loaded_model.predict(X_test)