Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # iterowanie po elementach ze zbioru test
- for index_test, row_test in test.iterrows():
- distances_3 = [] #tablica zawierajaca odleglosci do klasy 3
- distances_5 = [] #tablica zawierajaca odleglosci do klasy 5
- #iterowanie po elementach ze zbioru train
- for index_train, row_train in train.iterrows():
- if (row_train[0]==3):
- distances_3.append((distance(row_test[3], row_test[10], row_train[3], row_train[10])))
- else:
- distances_5.append((distance(row_test[3], row_test[10], row_train[3], row_train[10])))
- np_distances_3 = np.array(distances_3)
- np_distances_5 = np.array(distances_5)
- print("dystanse do klasy 3")
- print(np_distances_3)
- print("dystanse do klasy 5")
- print(np_distances_5)
- if(np_distances_3.min()>np_distances_5.min()):
- if(row_test[0]==5):
- print("Zaklasyfikowano poprawnie")
- else:
- print("Zaklasyfikowano niepoprawnie")
- else:
- if(row_test[0]==3):
- print("Zaklasyfikowano poprawnie")
- else:
- print("Zaklasyfikowano niepoprawnie")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement