veranoscience's picture
Space export: API + Dockerfile (model loaded from HF Hub)
9f0dbb9
raw
history blame contribute delete
262 Bytes
"""Point d'entrée : entraîne et sauvegarde le modèle final."""
from src.train_model import train_and_save
if __name__ == "__main__":
f1 = train_and_save()
print(f"[OK] Modèle entraîné et sauvegardé dans models/model.joblib | F1_test = {f1:.3f}")