predict_minimal.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
predict_minimal.py
------------------
Eğitilmiş minimal model ile tek bir e-postayı puanlar ve süreyi ölçer.
Kullanım:
python predict_minimal.py --subject "başlık" --text "gövde" --model spam_model_min.joblib
"""

import argparse
import joblib
import time

def main():
    ap = argparse.ArgumentParser()
    ap.add_argument("--subject", required=True, help="E-posta başlığı")
    ap.add_argument("--text", required=True, help="E-posta gövdesi")
    ap.add_argument("--model", default="spam_model_min.joblib", help="Model dosyası")
    ap.add_argument("--threshold", type=float, default=0.65, help="Spam eşiği (0-1)")
    args = ap.parse_args()

    # Modeli yükle
    pipe = joblib.load(args.model)

    # Subject + text birleştir
    text = (args.subject or "") + " \n " + (args.text or "")

    # Zaman ölçümü başla
    start_time = time.time()

    # Tahmin
    if hasattr(pipe, "predict_proba"):
        p = pipe.predict_proba([text])[:,1][0]
        etiket = int(p >= args.threshold)
        result = f"Olasılık (spam): {p:.4f} | Eşik: {args.threshold:.2f} | Tahmin: {'SPAM' if etiket else 'NORMAL'}"
    else:
        y = pipe.predict([text])[0]
        result = f"Tahmin: {'SPAM' if int(y)==1 else 'NORMAL'} (Not: olasılık desteği yok)"

    # Zaman ölçümü bitir
    elapsed = time.time() - start_time

    print(result)
    print(f"Geçen süre: {elapsed:.4f} saniye")

if __name__ == "__main__":
    main()