x_4, y_4 = DataGenerator.chessboard(N, 4, 0.04)
x_5, y_5 = DataGenerator.chessboard(N, 5, 0.04)
data_to_test = list()
data_to_test.append((x_1, y_1))
data_to_test.append((x_2, y_2))
data_to_test.append((x_3, y_3))
data_to_test.append((x_4, y_4))
data_to_test.append((x_5, y_5))
efficiency_list = []
for i in range(len(data_to_test)):
x = data_to_test[i][0]
y = data_to_test[i][1]
PlotGenerator.show_generated_data(x, y)
teach_data, test_data, teach_result, test_result = DataManager.split_data(
x, y, 0.5)
vec = range(1, len(teach_data), STEP)
efficiency = []
for k in vec:
neighbour_efficiency = 0
print("Iteration %d" % k)
for i in range(REPETITION):
kNeighborsClassifier = KNeighborsClassifier(k)
kNeighborsClassifier.fit(teach_data, teach_result)
neighbour_efficiency += kNeighborsClassifier.score(
test_data, test_result)
neighbour_efficiency /= REPETITION
efficiency.append((k, neighbour_efficiency))
efficiency_list.append(efficiency)
