Compare commits

..

2 Commits

Author SHA1 Message Date
279f4b82f7 ignore: csv and joblib files 2026-07-30 17:48:47 +03:00
2ae6193b03 fix: properly save model 2026-07-30 17:48:30 +03:00
2 changed files with 4 additions and 3 deletions

2
.gitignore vendored
View File

@@ -12,3 +12,5 @@ wheels/
.conda .conda
/lab2/*.pt /lab2/*.pt
*.safetensors *.safetensors
*.joblib
*.csv

View File

@@ -65,11 +65,10 @@ def main():
for text, label, score in zip(samples, preds, scores_demo): for text, label, score in zip(samples, preds, scores_demo):
verdict = "SPAM" if label == 1 else "HAM" verdict = "SPAM" if label == 1 else "HAM"
print(f"[{verdict}] score={score:.2f} :: {text}") print(f"[{verdict}] score={score:.2f} :: {text}")
if __name__ == "__main__":
# persist our model # persist our model
joblib.dump(vectorizer, "vectorizer.joblib") joblib.dump(vectorizer, "vectorizer.joblib")
joblib.dump(clf, "classifier.joblib") joblib.dump(clf, "classifier.joblib")
if __name__ == "__main__":
print("\nSaved model artifacts.") print("\nSaved model artifacts.")
main() main()