bajiang's picture
Initial commit: upload model and source code
e4b734c
raw
history blame contribute delete
263 Bytes
import joblib
import numpy as np
# Load the trained model
model = joblib.load("random_forest_model.pkl")
def predict(features):
"""features: list of 24 numeric values"""
features = np.array(features).reshape(1, -1)
return model.predict(features)[0]