kalimdor_knn_classifier_training_and_prediction.ts

1import { KNN } from 'kalimdor';
2
3// Sample dataset: [feature1, feature2]
4const X = [
5  [1, 2],
6  [2, 3],
7  [3, 4],
8  [10, 11],
9  [11, 12],
10  [12, 13],
11];
12
13// Corresponding labels
14const y = [0, 0, 0, 1, 1, 1];
15
16// Initialize the KNN classifier with k=3
17const knn = new KNN({ k: 3 });
18
19// Train the model
20knn.fit(X, y);
21
22// Make a prediction for a new data point
23const prediction = knn.predict([[2.5, 3.5]]);
24
25console.log(`Prediction: ${prediction}`); // Output: [0]