0

您好,我在下面编写了简单的代码来探索模糊 Cmean 聚类

import pandas as pd
import numpy as np
from os import listdir
from sklearn.model_selection import train_test_split
from skfuzzy.cluster import cmeans, cmeans_predict
from sklearn.metrics import classification_report,confusion_matrix

def find_csv_filenames( path_to_dir, suffix=".csv" ):
    filenames = listdir(path_to_dir)
    return [ path_to_dir+filename for filename in filenames if filename.endswith( suffix ) ]

listFiles = find_csv_filenames('<Path to folder with csv files>')
for files in listFiles:
    df = pd.read_csv(files)
    df.loc[df['bug']>1,'bug']=1
    df2 =df.iloc[:,3:]
    #Above are some pre processing steps
    #Below splitting data for test and train
    X_train, X_test = train_test_split(df2, test_size=0.30)
    #dropping bug column for unsupervised learning
    X_train2 = X_train.drop('bug',axis=1) 
    X_test2  = X_test.drop('bug',axis=1) 
    print (X_train2.shape)
    #Shape is 163,20 for 163 training data with 20 features
    cntr, u, u0, d, jm, p, fpc = cmeans(X_train2,2,2,0.25,500,init=None, seed=None)
    print(cntr.shape)
    #above shape is coming 2,163

来自上述 cmeam 算法的中心的大小为(2,163)但由于我的训练数据只有 20 个特征,因此 cntr 的形状应该是 (2,20)。无法理解我错在哪里

4

1 回答 1

3

skfuzzy文档中:

数据:二维数组,大小(S,N)

要聚类的数据。N是数据集的数量;S 是每个样本向量内的特征数。

因此,您需要转置您的输入,但未经测试:

cmeans(X_train2.T, ...)

应该管用。

于 2018-04-27T07:37:52.317 回答