LeonardoMdSA/Context-aware-NLP-classification-platform-with-MCP

1#scripts\train_model.py2from pathlib import Path3import sys4 5# Add project root to sys.path6sys.path.append(str(Path(__file__).resolve().parent.parent))7 8import joblib9from app.classification.sklearn_model import SklearnClassifier10 11# -------------------------12# Paths13# -------------------------14DATASET_PATH = Path(__file__).parent.parent / "data" / "samples" / "training_data.json"15MODEL_PATH = Path(__file__).parent.parent / "models" / "trained_pipeline.joblib"16 17# -------------------------18# Train classifier19# -------------------------20print(f"Loading training data from {DATASET_PATH}")21classifier = SklearnClassifier(dataset_path=str(DATASET_PATH))22 23# Save trained pipeline24joblib.dump(classifier.pipeline, MODEL_PATH)25print(f"Trained model saved to {MODEL_PATH}")26