Guest User

Untitled

a guest
Jan 18th, 2018
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.50 KB | None | 0 0
  1. dataset = pd.read_csv(file)
  2. training_samples = get_sample_number(dataset)
  3. X_train = dataset.iloc[:training_samples, 5:9]
  4. y_train = dataset.iloc[:training_samples, 9]
  5. X_test = dataset.iloc[training_samples:, 5:9]
  6.  
  7. # Feature Scaling
  8. sc = StandardScaler()
  9. X_train = sc.fit_transform(X_train)
  10. X_test = sc.fit_transform(X_test)
  11.  
  12. # Fitting classifier to the training set
  13. classifier = KNeighborsClassifier(n_neighbors=5, metric='minkowski', p=2)
  14. classifier.fit(X_train, y_train)
  15.  
  16. y_pred = classifier.predict(X_test)
Add Comment
Please, Sign In to add comment