From 6eebe68539a2c1b05ffddb8dbcd9fc44582d1bcc Mon Sep 17 00:00:00 2001
From: Nico Adamo <nadamo@caltech.edu>
Date: Mon, 6 Nov 2023 17:37:33 -0800
Subject: [PATCH] Add more docstring info
---
NaiveBayes.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/NaiveBayes.py b/NaiveBayes.py
index ffbe6969..79289898 100644
--- a/NaiveBayes.py
+++ b/NaiveBayes.py
@@ -1,8 +1,7 @@
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
--
GitLab