We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
BaseEstimator
LinearModel
1 parent c099fc4 commit c35bc67Copy full SHA for c35bc67
pymc_extras/linearmodel.py
@@ -2,10 +2,12 @@
2
import pandas as pd
3
import pymc as pm
4
5
+from sklearn.base import BaseEstimator
6
+
7
from pymc_extras.model_builder import ModelBuilder
8
9
-class LinearModel(ModelBuilder):
10
+class LinearModel(ModelBuilder, BaseEstimator):
11
def __init__(
12
self, model_config: dict | None = None, sampler_config: dict | None = None, nsamples=100
13
):
0 commit comments