34e7e918bd4fc09f643e8bb448e875a9bcb7e841,experiment.py,,,#,4

Before Change

    joined["y"] = data.y
    mod = model.from_formula(formula, joined)
    res = mod.fit()
    print(res)

After Change

import numpy as np
y = np.arange(12.0)[:,None]
import pandas as pd
entities = pd.Categorical(pd.Series(["a"]*6+["b"]*6))
dummies = pd.get_dummies(entities)
w = np.random.chisquare(5, (12,1)) / 5
w = w/w.mean()
root_w = np.sqrt(w)
wd = root_w  * dummies.values
wy = root_w  * y
b = np.linalg.pinv(wd) @ wy

Instances