Machine_learning_2023/decisionTree/evaluate.py

9 lines
207 B
Python

import joblib
def evaluate(data):
# Load the model
clf = joblib.load('decisionTree/decision_tree_model.pkl')
# Make a prediction
prediction = clf.predict(data)
return prediction