#9 add part of enhance classifier with request

thegodenage · May 16, 2024 · 2d6d680 · 2d6d680
1 parent a4cde35
commit 2d6d680
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internal/ddosml/knn.go b/internal/ddosml/knn.go
@@ -23,8 +23,10 @@ func NewKNNClassifier() *KnnClassifier {
 }
 
 func (k *KnnClassifier) EnhanceClassifierWithRequest(m *Request) {
-	//TODO implement me
-	panic("implement me")
+	k.mu.Lock()
+	defer k.mu.Unlock()
+
+	k.knn.UpdateTrainingSet()
 }
 
 func (k *KnnClassifier) IsRequestPotentialDDOS(ctx context.Context, m *Request) bool {