diff --git a/river/base/__init__.py b/river/base/__init__.py index 2d6f2c2c97..a92e437b32 100644 --- a/river/base/__init__.py +++ b/river/base/__init__.py @@ -15,6 +15,7 @@ from __future__ import annotations from . import tags, typing +from .adpredictor import AdPredictor from .base import Base from .classifier import Classifier, MiniBatchClassifier from .clusterer import Clusterer @@ -35,7 +36,6 @@ Transformer, ) from .wrapper import Wrapper -from .adpredictor import AdPredictor __all__ = [ "AdPredictor", diff --git a/river/base/adpredictor.py b/river/base/adpredictor.py index bcd5deb49f..e95b581ffd 100644 --- a/river/base/adpredictor.py +++ b/river/base/adpredictor.py @@ -10,6 +10,7 @@ def default_weight(): return {"mean": 0.0, "variance": 1.0} + class AdPredictor(Classifier): """AdPredictor, developed by Microsoft, is a machine learning algorithm designed to predict the probability of user clicks on online advertisements. This algorithm plays a crucial role in computational advertising, where predicting @@ -66,7 +67,6 @@ def train_and_test(model, data): config = namedtuple("config", ["beta", "prior_probability", "epsilon", "num_features"]) - def __init__(self, beta=0.1, prior_probability=0.5, epsilon=0.1, num_features=10): # Initialization of model parameters self.beta = beta diff --git a/river/linear_model/__init__.py b/river/linear_model/__init__.py index 33d6cfa05e..756720490a 100644 --- a/river/linear_model/__init__.py +++ b/river/linear_model/__init__.py @@ -11,8 +11,6 @@ from .perceptron import Perceptron from .softmax import SoftmaxRegression - - __all__ = [ "base", "ALMAClassifier", @@ -23,6 +21,4 @@ "PARegressor", "Perceptron", "SoftmaxRegression", - - ]