#!/usr/bin/env python # -*- coding: UTF-8 -*- import numpy as np from sklearn import datasets, linear_model dataPath = r"./multipleDummy.csv" deliveryData = np.genfromtxt(dataPath, delimiter=',') print "data" print deliveryData X = deliveryData[:, :-1] Y = deliveryData[:, -1] print "X:", X print "Y:", Y regr = linear_model.LinearRegression() regr.fit(X, Y) print "coefficients:", regr.coef_ print "intercept:", regr.intercept_ # 转换成一维数组,否则执行predict函数时报错 xPred = np.array([102, 6, 0, 1, 0]) yPred = regr.predict(xPred.reshape(1, 5)) # print yPred