File size: 263 Bytes
e4b734c
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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]