File size: 403 Bytes
379e9a4
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

from sklearn.linear_model import LinearRegression
import numpy as np

def predict(features: dict) -> tuple[float, dict]:
    model = LinearRegression()
    X = np.array(list(features.values())).reshape(1, -1)
    pred = float(model.predict(X)) if hasattr(model, 'coef_') else 0.0
    metrics = {
        "r_squared": 0.0,
        "coefficients": {},
        "t_stats": {}
    }
    return pred, metrics