INNER CODE UNIT · Python
train_model
bansalkanav/Ultimate-Data-Science-Toolkit---From-Python-Basics-to-GenerativeAI · Module 5 - MLOPs/4. Orchestrate ML Pipeline/my_workflow_script.py:49
def train_model(X_train_scaled, y_train, hyperparameters):
"""
Training the machine learning model.
"""
clf = KNeighborsClassifier(**hyperparameters)
clf.fit(X_train_scaled, y_train)
return clf
@task
def evaluate_model(model, X_train_scaled, y_train, X_test_scaled, y_test):
"""
Evaluating the model.
"""
y_train_pred = model.predict(X_train_scaled)
y_test_pred = model.predict(X_test_scaled)
train_score = metrics.accuracy_score(y_train, y_train_pred)