File size: 421 Bytes
588124b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from sklearn.ensemble import RandomForestClassifier
def train_model(X_train, y_train):
"""
Train a Random Forest Classifier.
Args:
X_train (pd.DataFrame): Training features.
y_train (pd.Series): Training labels.
Returns:
model: Trained Random Forest model.
"""
model = RandomForestClassifier(random_state=42)
model.fit(X_train, y_train)
return model
|