没有for循环,它变得非常容易。
import numpy as np from sklearn import decomposition from sklearn import linear_model features_train = np.loadtxt('C:\\Users\\bruker\\Desktop\\Data\\CFD_features.txt', delimiter=',') ratings_train = np.loadtxt('C:\\Users\\bruker\\Desktop\\Data\\CFD_ratings.txt', delimiter=',') pca = decomposition.PCA(n_components=13) pca.fit(features_train) features_train = pca.transform(features_train) regr = linear_model.LinearRegression() regr.fit(features_train, ratings_train) features_test = np.loadtxt('C:\\Users\\bruker\\Desktop\\Data\\CFD_features_Test.txt', delimiter=',') features_test = pca.transform(features_test) predictions = regr.predict(features_test)