from sklearn.datasets import load_boston from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression import time boston = load_boston() X = boston.data y = boston.target # 将数据集分成80%的训练集和20%的测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=3) model = LinearRegression(normalize=True) # 计算建模时间 start = time.perf_counter() model.fit(X_train, y_train) time_score = time.perf_counter() - start; train_score = model.score(X_train, y_train) cv_score = model.score(X_test, y_test) print("elaspe: {0:.6f}; train_score: {1:.6f}; cv_score:{2:.6f}".format(time_score, train_score, cv_score))
sklearn----LinearRegresion
最新推荐文章于 2022-04-09 12:03:02 发布