sklearn-PCA降维以及保存模型

对训练集使用fit_transform函数进行学习

# 已知训练集train、验证集dev和测试集test
from sklearn.decomposition import PCA 
pca = PCA(n_components=2)	# 特征个数
# 对训练集使用fit_transform函数进行学习
X_train = pca.fit_transform(X_train)

对dev、test使用transform函数,进行测试

# 对dev、test使用transform函数,进行测试
X_dev = pca.transform(X_dev)

使用joblib保存、加载模型

# 保存模型--python 3.7
import joblib
joblib.dump(pca_model, pca_model_dir)

# 加载模型
pca_model = joblib.load(pca_model_dir)
X_test = pca_model.transform(X_test)
...

猜你喜欢

转载自blog.csdn.net/tailonh/article/details/112404197
今日推荐