worldcup-api / train.py
wihaha's picture
First release to Hugging Face
a3d10df
Raw
History Blame Contribute Delete
469 Bytes
import os
import joblib
from rule_based_model import RuleBasedModel
def train_baseline_model():
print("Initializing Rule-Based Baseline Model (avoiding time leakage from current matches)...")
clf = RuleBasedModel()
# 6. Save Model
os.makedirs('artifacts', exist_ok=True)
joblib.dump(clf, 'artifacts/rule_based_baseline.pkl')
print("Model saved to artifacts/rule_based_baseline.pkl")
if __name__ == "__main__":
train_baseline_model()