diff --git a/main.py b/main.py index 36b9314..9dbe4d0 100644 --- a/main.py +++ b/main.py @@ -2,6 +2,13 @@ Autor: Dominik Strzałko Data: 05.08.2021 Zadanie: naiwny bayes2 gotowa biblioteka (Skeptic vs paranormal subreddits) + +Wyniki z geval: +Likelihood 0.0000 +Accuracy 0.7367 +F1.0 0.4367 +Precision 0.8997 +Recall 0.2883 ''' import numpy as np from sklearn.preprocessing import LabelEncoder @@ -51,7 +58,7 @@ def predict(model, X_tsv, file_name): def main(): - + model = Create_model("train/in.tsv", "train/expected.tsv") predict(model, "dev-0/in.tsv", "dev-0/out.tsv")