File size: 469 Bytes
a3d10df
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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()