首页 > 其他 > 详细

机器学习基础功能练习II

时间:2020-07-07 13:11:04      阅读:61      评论:0      收藏:0      [点我收藏+]

机器学习基础功能练习II

一、导入sklearn 数据集

  

from sklearn.datasets import load_diabetes
diabetes = load_diabetes()
"""返回字典,数据集的descr,data,feature_names等关键数据
diabetes.data 是一个矩阵
sklearn.datasets.load_boston
sklearn.datasets.load_breast_cancer
sklearn.datasets.load_diabetes
sklearn.datasets.load_digits
sklearn.datasets.load_files
sklearn.datasets.load_iris
"""
print(diabetes)
print(diabetes.feature_names)
print(diabetes.data)
X = diabetes.data
y = diabetes.target

  

二、分隔数据集

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=1)
test_size=0.3 代表7:3,random_state=1 0~42

  

三、模型训练与预测

from sklearn.linear_model import LinearRegression
model = LinearRegression(copy_X=True, fit_intercept=True, n_jobs=1, normalize=False)

model.fit(X_train, y_train)
y_predict = model.predict(X_test)

  

机器学习基础功能练习II

原文:https://www.cnblogs.com/lx63blog/p/13260142.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!