Final Version (+ geval)
This commit is contained in:
parent
48a3c4eace
commit
71104493a9
9
main.py
9
main.py
@ -2,6 +2,13 @@
|
|||||||
Autor: Dominik Strzałko
|
Autor: Dominik Strzałko
|
||||||
Data: 05.08.2021
|
Data: 05.08.2021
|
||||||
Zadanie: naiwny bayes2 gotowa biblioteka (Skeptic vs paranormal subreddits)
|
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
|
import numpy as np
|
||||||
from sklearn.preprocessing import LabelEncoder
|
from sklearn.preprocessing import LabelEncoder
|
||||||
@ -51,7 +58,7 @@ def predict(model, X_tsv, file_name):
|
|||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
|
|
||||||
model = Create_model("train/in.tsv", "train/expected.tsv")
|
model = Create_model("train/in.tsv", "train/expected.tsv")
|
||||||
|
|
||||||
predict(model, "dev-0/in.tsv", "dev-0/out.tsv")
|
predict(model, "dev-0/in.tsv", "dev-0/out.tsv")
|
||||||
|
Loading…
Reference in New Issue
Block a user