sklearn.externals import joblib模块保存和下载使用模型的用法实例

 #加载模块
 from sklearn import datasets
 from sklearn.externals import joblib
 from sklearn.linear_model import LinearRegression
 from sklearn.cross_validation import train_test_split
 #分割数据集
 data_x,data_y = datasets.load_iris(return_X_y=True)
 train_X,test_X,train_y,test_y = train_test_split(data_x,data_y,test_size=0.2,random_state=2,stratify=data_y)
 #训练模型
 lr = LinearRegression()
 lr.fit(train_X,train_y)

 #将训练的模型保存
 direction = joblib.dump(lr,"H:/lr_model_20180911.pkl")

 #下载模型
 lr = joblib.load("".join(direction))
 #模型预测
 lr.predict(test_X)
 #重新设置模型参数并训练
 lr.set_params(normalize=True).fit(train_X,train_y)
 #新模型做预测
 lr.predict(test_X)