check_model.py 210 Bytes
Newer Older
1
2
3
4
5
6
from sissopp import ModelLogRegressor
from pathlib import Path

import numpy as np

model = ModelLogRegressor(
7
    str("models/train_dim_2_model_0.dat")
8
9
10
)
assert model.rmse < 1e-4
assert model.test_rmse < 1e-4