feat: functional KNN

c0utin · Sep 30, 2024 · 62f8cec · 62f8cec
1 parent 975dd1a
commit 62f8cec
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/parsekoto_backend/main.mo b/src/parsekoto_backend/main.mo
@@ -82,7 +82,7 @@ actor {
 		return studytimeDiff + absencesDiff;
 	};
 
-	public func calculateAllDistances(newRecord: Types.StudentRecord): async [(Types.StudentRecord, Float)] {
+	public func calculateAllDistancesMock(newRecord: Types.StudentRecord): async [(Types.StudentRecord, Float)] {
 		var distances: [(Types.StudentRecord, Float)] = [];
 
 		for (mockRecord in mock.vals()) { 
@@ -93,5 +93,17 @@ actor {
 		return distances;
 	};
 
+
+	public func calculateAllDistances(newRecord: Types.StudentRecord): async [(Types.StudentRecord, Float)] {
+		var distances: [(Types.StudentRecord, Float)] = [];
+
+		for (realRecord in stable_decoded_student.vals()) { 
+			let distance = await calculateDistance(newRecord, realRecord);
+			distances := Array.append(distances, [(realRecord, distance)]);
+		};
+
+		return distances;
+	};
+
 };