Add more docstring info

6eebe685 · Nicholas Stone Adamo · d4e1a604 · 6eebe685
Commit 6eebe685 authored 1 year ago by Nicholas Stone Adamo
Hide whitespace changes
Inline Side-by-side

Showing

with 3 additions and 2 deletions
+3 -2
--- a/NaiveBayes.py
+++ b/NaiveBayes.py
 import os
 import math
 from typing import List
-a = 1
-k = 1
+k = 1 # Laplace smoothing constant

 class NaiveBayes:
    def __init__(self, data_folder, train_subfolder, validate_subfolder, c1_folder, c2_folder):
@@ -17,6 +16,8 @@ class NaiveBayes:
        pass

    # TODO: Implement Naive Bayes classification
+    # Should write output to classification.txt in the form "file.txt classification"
+    # where classification is one of c1_folder or c2_folder
    def classify(self, files: List[str], folder: str):
        pass