def visualise(self, data):
mapper = UMAPBase()
mapper.embedding_ = data[["x", "y"]].values
p = interactive(mapper,
labels=data["class"],
hover_data=data,
point_size=5,
interactive_text_search=False)
del p.tools[len(p.tools) - 1]
p.add_tools(get_hovertool())
p.add_tools(LassoSelectTool())
p_json = json_item(p)
return p_json
def test_umap_transform_on_iris_modified_dtype(iris, iris_selection):
data = iris.data[iris_selection]
fitter = UMAP(n_neighbors=10, min_dist=0.01, random_state=42).fit(data)
fitter.embedding_ = fitter.embedding_.astype(np.float64)
new_data = iris.data[~iris_selection]
embedding = fitter.transform(new_data)
trust = trustworthiness(new_data, embedding, 10)
assert_greater_equal(
trust,
0.8,
"Insufficiently trustworthy transform for iris dataset: {}".format(
trust),
)
