def test_save_load():
""" tests saving and loading
"""
X, y = make_moons(100)
embedder = ParametricUMAP()
embedding = embedder.fit_transform(X)
embedder.save("/tmp/model")
embedder = load_ParametricUMAP("/tmp/model")
def test_save_load():
"""tests saving and loading"""
X, y = make_moons(100)
embedder = ParametricUMAP()
embedding = embedder.fit_transform(X)
# if platform.system() != "Windows":
# Portable tempfile
model_path = tempfile.mkdtemp(suffix="_umap_model")
embedder.save(model_path)
embedder = load_ParametricUMAP(model_path)
def test_save_load(moon_dataset):
"""tests saving and loading"""
embedder = ParametricUMAP()
embedding = embedder.fit_transform(moon_dataset)
# completes successfully
assert embedding is not None
assert embedding.shape == (moon_dataset.shape[0], 2)
# if platform.system() != "Windows":
# Portable tempfile
model_path = tempfile.mkdtemp(suffix="_umap_model")
embedder.save(model_path)
loaded_model = load_ParametricUMAP(model_path)
assert loaded_model is not None
def test_save_load(moon_dataset):
"""tests saving and loading"""
embedder = ParametricUMAP()
embedding = embedder.fit_transform(moon_dataset)
# completes successfully
assert embedding is not None
assert embedding.shape == (moon_dataset.shape[0], 2)
# Portable tempfile
model_path = tempfile.mkdtemp(suffix="_umap_model")
embedder.save(model_path)
loaded_model = load_ParametricUMAP(model_path)
assert loaded_model is not None
loaded_embedding = loaded_model.transform(moon_dataset)
assert_array_almost_equal(
embedding,
loaded_embedding,
decimal=5,
err_msg="Loaded model transform fails to match original embedding",
)
