我正在尝试使用0.15.2版本的scikit-learn。在此版本中,文档显示单独的fit(X)
和predict(X)
函数以及先前版本中提供的组合fit_predict(X)
函数。
我认为这可以让我适应一个阵列然后预测一个新阵列(所以fit(X)
和predict(Y)
)。但是,当我尝试这样做时,由于尺寸不匹配,我得到ValueError
。
这是我的代码:
from sklearn.cluster import MeanShift, estimate_bandwidth
bandwidth = estimate_bandwidth(X, quantile=0.2, n_samples=5000) # n_samples 100000 memory bound
ms = MeanShift(bandwidth=bandwidth)
ms.fit(X)
results = ms.predict(Y)
有没有办法使用模型预测新(不同)数组的簇而不是最初拟合的数组?
我发现了这篇文章(MeanShift `fit` vs `fit_predict` scikitlearn),但我认为它提出了一个与我想要提出的问题不同的问题。