Skip to content

Commit c35bc67

Browse files
Add BaseEstimator mixin to LinearModel (#519)
1 parent c099fc4 commit c35bc67

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pymc_extras/linearmodel.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,12 @@
22
import pandas as pd
33
import pymc as pm
44

5+
from sklearn.base import BaseEstimator
6+
57
from pymc_extras.model_builder import ModelBuilder
68

79

8-
class LinearModel(ModelBuilder):
10+
class LinearModel(ModelBuilder, BaseEstimator):
911
def __init__(
1012
self, model_config: dict | None = None, sampler_config: dict | None = None, nsamples=100
1113
):

0 commit comments

Comments
 (0)