资源简介

线性回归实现sklearn自带的糖尿病datasets数据集

资源截图

代码片段和文件信息

# -*- coding: utf-8 -*-
“““
Created on Mon Oct  8 09:13:23 2018

@author:xxh
“““
from sklearn import datasets       #糖尿病数据集
from sklearn import linear_model   #sklearn的线性回归算法
import matplotlib.pyplot as plt    #绘图
import numpy as np
#数据集获取
diabetes = datasets.load_diabetes()        #载入数据 
diabetes_x = diabetes.data[: np.newaxis]  #获取一个特征

#数据集划分
diabetes_x_temp = diabetes_x[: : 2] 
diabetes_x_train = diabetes_x_temp[:-20]   #训练样本
diabetes_x_test = diabetes_x_temp[-20:]    #测试样本 后20行
diabetes_y_train = diabetes.target[:-20]   #训练标记
diabetes_y_test = diabetes.target[-20:]    #预测对比标记 
print(u‘划分行数:‘ “[总数据量]“len(diabetes_x_temp)“   [训练集]“ len(diabetes_x_train)“   [测试集]“ len(diabetes_x_test))
print(diabetes_x_test)              

评论

共有 条评论

相关资源