У меня есть следующая программа, я хотел бы вывести свои данные на график, не могли бы вы показать мне, как это сделать?
У меня есть поиск и попытка, но он не работает.
import csv
import numpy as np
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import confusion_matrix, f1_score, accuracy_score
def read_iris(fname):
with open(fname, 'r') as f:
data = list(csv.reader(f))[:-1]
xs = [[float(s) for s in d[:-1]] for d in data]
ys = [ d[-1] for d in data]
# for d in data[:5]: print(d)
# for x in xs[:5]: print(x)
# for y in ys[:5]: print(y)
return xs, ys
def read_bank(fname):
with open('./data/output.csv', 'r') as f:
data = list(csv.reader(f))
labels = data[0]
data = data[1:]
xs = [d[1:-2] for d in data]
ys = [d[-2:] for d in data]
xs = [[float(i) for i in x] for x in xs]
ys = ['yes' if y[1] == '1' else 'no' for y in ys]
return xs, ys
def main():
xs, ys = read_bank('./data/output.csv')
xs_train, xs_test, ys_train, ys_test = train_test_split(xs, ys, test_size=0.3)
# train & predict labele
knn = KNeighborsClassifier(n_neighbors=10)
knn.fit(xs_train, ys_train)
ps = knn.predict(xs_test)
ac = accuracy_score(ps, ys_test)
print(ac)
if __name__ == '__main__': main()
Я использовал функцию plt.plot, но я не знаю, какие именно параметры внутри (plt.plot (--------)).
Спасибо за мое плохое понимание, потому что я новичок.