【发布时间】:2019-05-23 03:07:57
【问题描述】:
我正在学习如何使用 make_pipeline 运行 K-means 模型来标准化我的数据集列的值。
我正在学习 DataCamp 课程,但我不清楚为什么它们在同一数据集上拟合和预测模型 - 在 Datacamp 案例“运动”中,即每日股票价值数据集。我认为 K-means 模型的全部目的是在训练数据集上进行训练并预测测试数据集?
与 Datacamp 案例不同,我想在列标准化的训练数据集上训练我的模型,并在列标准化的测试数据集上对其进行测试。怎么做?我正在复制并粘贴下面的 Datacamp 代码以供参考。
# Import Normalizer
from sklearn.preprocessing import Normalizer
# Create a normalizer: normalizer
normalizer = Normalizer()
# Create a KMeans model with 10 clusters: kmeans
kmeans = KMeans(n_clusters = 5)
# Make a pipeline chaining normalizer and kmeans: pipeline
pipeline = make_pipeline(normalizer, kmeans)
# Fit pipeline to the daily price movements
pipeline.fit(movements)
# Predict the cluster labels: labels
labels = pipeline.predict(movements)
【问题讨论】:
标签: python pandas scikit-learn k-means