Download src/train_classifier.py from WizardCoder2007/social_media_analyzer: direct link, hf CLI and curl.
- Browser
- Download file 401 Bytes
-
https://huggingface.co/spaces/WizardCoder2007/social_media_analyzer/resolve/main/src/train_classifier.py
- Command line
-
hf download hf://spaces/WizardCoder2007/social_media_analyzer/src/train_classifier.py
-
curl -L -o train_classifier.py https://huggingface.co/spaces/WizardCoder2007/social_media_analyzer/resolve/main/src/train_classifier.py
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") | |