模糊 C 均值中的 ValidationError

Question

我尝试在 jupyter notebook 和 colab 中运行，但仍然收到 fcmeans 的此错误。 但它在不同的笔记本电脑上运行良好。 这是用于分割数据集的代码

# Stratified Sampling using Scikit-learn's Stratified Shuffle Split Class
from sklearn.model_selection import StratifiedShuffleSplit

split = StratifiedShuffleSplit(n_splits=1, test_size=0.25, random_state=42)

for train_index, test_index in split.split(data1, data1["class"]):
    strat_train_set = data1.loc[train_index]

    strat_test_set = data1.loc[test_index]
    train_set = strat_train_set.drop("class", axis=1) # drop labels for training set

train_labels = strat_train_set["class"].copy()
test_set = strat_test_set.drop("class", axis=1) # drop labels for testing set
test_labels = strat_test_set["class"].copy()

那我错过了什么？

Answer 1

这里的问题是，tr_set 不是numpy.ndarray。因此，您需要做的就是将数据帧作为 numpy 数组传递。

在您的情况下，如果在将数据传递给fit 之前使用to_numpy 函数（例如fcm.fit(tr_set.to_numpy())），它将起作用。

这在 fcm 文档中非常清楚。