pyodide/packages/river/test_river.py

18 lines
478 B
Python

import pytest
from pytest_pyodide import run_in_pyodide
@pytest.mark.driver_timeout(60)
@run_in_pyodide(packages=["river"])
def test_linear_regression(selenium):
from river import datasets, evaluate, linear_model, metrics, preprocessing
dataset = datasets.TrumpApproval()
model = preprocessing.StandardScaler() | linear_model.LinearRegression(
intercept_lr=0.1
)
metric = metrics.MAE()
evaluate.progressive_val_score(dataset, model, metric)