# 输入CSV文件,输出平均准确率
import pandas as pd
import numpy as np
from sklearn.model_selection import KFold
from sklearn.metrics import accuracy_score
from sklearn_lvq import LgmlvqModel
if __name__ == '__main__':
dataset = np.array(pd.read_csv("ionosphere-full.csv", sep=',', header=None))
k_Cross = KFold(n_splits=5, random_state=8, shuffle=True)
index = 0
score = np.array([])
data,label = dataset[:,:-1],dataset[:,-1]
for train_index, test_index in k_Cross.split(dataset):
train_data, train_label = data[train_index, :], label[train_index]
test_data, test_label = data[test_index, :], label[test_index]
model = LgmlvqModel()
model.fit(train_data,train_label)
pred = model.predict(test_data)
acc = accuracy_score(test_label, pred)
score = np.append(score,acc)
print('score[{}] = {}%'.format(index,acc))
index+=1
print('mean_accuracy = {}%'.format(np.mean(score)))