import numpy as np
from mosaic_online import MOSAIC

rng = np.random.default_rng(0)
stream = rng.standard_normal((1000, 64))

model = MOSAIC(n_features=64, rank=8, eta_u=0.08, eta_s=0.03)
for x in stream:
    model.partial_fit(x)

print("coords:", model.transform_one(stream[-1]).shape)
print("novelty:", model.novelty_score(stream[-1]))
print("fixed state floats:", model.memory_floats())