social_media_analyzer / src /train_classifier.py
WizardCoder2007's picture
update
bbd259b
raw
history blame contribute delete
401 Bytes
import joblib
from sklearn.linear_model import LogisticRegression
print("train_classifier module loaded")
def train_and_save(X, y):
"""
Train final stance classifier and save it
"""
clf = LogisticRegression(
max_iter=2000,
multi_class="multinomial"
)
clf.fit(X, y)
joblib.dump(clf, "models/final_classifier.pkl")
print("✅ Model trained and saved")