Compare commits
2 Commits
0d10a278bf
...
279f4b82f7
| Author | SHA1 | Date | |
|---|---|---|---|
| 279f4b82f7 | |||
| 2ae6193b03 |
2
.gitignore
vendored
2
.gitignore
vendored
@@ -12,3 +12,5 @@ wheels/
|
|||||||
.conda
|
.conda
|
||||||
/lab2/*.pt
|
/lab2/*.pt
|
||||||
*.safetensors
|
*.safetensors
|
||||||
|
*.joblib
|
||||||
|
*.csv
|
||||||
|
|||||||
@@ -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()
|
||||||
|
|||||||
Reference in New Issue
Block a user