Compare commits
2 Commits
0d10a278bf
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| 279f4b82f7 | |||
| 2ae6193b03 |
2
.gitignore
vendored
2
.gitignore
vendored
@@ -12,3 +12,5 @@ wheels/
|
||||
.conda
|
||||
/lab2/*.pt
|
||||
*.safetensors
|
||||
*.joblib
|
||||
*.csv
|
||||
|
||||
@@ -65,11 +65,10 @@ def main():
|
||||
for text, label, score in zip(samples, preds, scores_demo):
|
||||
verdict = "SPAM" if label == 1 else "HAM"
|
||||
print(f"[{verdict}] score={score:.2f} :: {text}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# persist our model
|
||||
joblib.dump(vectorizer, "vectorizer.joblib")
|
||||
joblib.dump(clf, "classifier.joblib")
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("\nSaved model artifacts.")
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user