linly/ai-sec-labs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: properly save model

Browse Source
This commit is contained in:
Aly Sewelam
2026-07-30 17:48:30 +03:00
parent 0d10a278bf
commit 2ae6193b03
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lab4/train_spam_model.py
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()