Spaces:
Running
Running
| 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() | |