chartmultiselect

%23%20%2F%2F%2F%20script%0A%23%20requires-python%20%3D%20%22%3E%3D3.10%22%0A%23%20dependencies%20%3D%20%5B%0A%23%20%20%20%20%20%22marimo%22%2C%0A%23%20%20%20%20%20%22matplotlib%22%2C%0A%23%20%20%20%20%20%22numpy%22%2C%0A%23%20%20%20%20%20%22scikit-learn%22%2C%0A%23%20%20%20%20%20%22wigglystuff%3D%3D0.2.31%22%2C%0A%23%20%5D%0A%23%20%2F%2F%2F%0A%0Aimport%20marimo%0A%0A__generated_with%20%3D%20%220.20.2%22%0Aapp%20%3D%20marimo.App()%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%20%20%20%20import%20matplotlib.pyplot%20as%20plt%0A%20%20%20%20import%20numpy%20as%20np%0A%20%20%20%20from%20sklearn.datasets%20import%20make_moons%2C%20make_blobs%0A%20%20%20%20from%20wigglystuff%20import%20ChartMultiSelect%0A%0A%20%20%20%20COLORS%20%3D%20%5B%22%231f77b4%22%2C%20%22%23ff7f0e%22%2C%20%22%232ca02c%22%2C%20%22%23d62728%22%5D%0A%20%20%20%20return%20COLORS%2C%20ChartMultiSelect%2C%20make_blobs%2C%20make_moons%2C%20mo%2C%20np%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(%22%22%22%0A%20%20%20%20%23%20ChartMultiSelect%20Demo%0A%0A%20%20%20%20Draw%20multiple%20box%20or%20lasso%20selections%20to%20classify%20points%20on%20a%20matplotlib%20chart.%0A%0A%20%20%20%20-%20Pick%20a%20**class**%20(a%2C%20b%2C%20...)%20then%20draw%20regions%20with%20**Box**%20or%20**Lasso**%0A%20%20%20%20-%20Click%20a%20selection%20to%20highlight%20it%2C%20then%20use%20the%20**delete**%20button%20to%20remove%20it%0A%20%20%20%20-%20Use%20**undo**%20to%20remove%20the%20last-drawn%20selection%0A%20%20%20%20-%20Use%20**clear**%20to%20remove%20all%20selections%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(make_moons%2C%20np)%3A%0A%20%20%20%20X_moons%2C%20y_moons%20%3D%20make_moons(n_samples%3D300%2C%20noise%3D0.15%2C%20random_state%3D42)%0A%20%20%20%20x_moons%20%3D%20X_moons%5B%3A%2C%200%5D%0A%20%20%20%20y_moons_coord%20%3D%20X_moons%5B%3A%2C%201%5D%0A%20%20%20%20%23%20Shift%20so%20the%20two%20moons%20are%20nicely%20centered%0A%20%20%20%20pad%20%3D%200.5%0A%20%20%20%20x_lo%2C%20x_hi%20%3D%20float(np.min(x_moons)%20-%20pad)%2C%20float(np.max(x_moons)%20%2B%20pad)%0A%20%20%20%20y_lo%2C%20y_hi%20%3D%20float(np.min(y_moons_coord)%20-%20pad)%2C%20float(np.max(y_moons_coord)%20%2B%20pad)%0A%20%20%20%20return%20x_hi%2C%20x_lo%2C%20x_moons%2C%20y_hi%2C%20y_lo%2C%20y_moons%2C%20y_moons_coord%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20COLORS%2C%0A%20%20%20%20ChartMultiSelect%2C%0A%20%20%20%20x_hi%2C%0A%20%20%20%20x_lo%2C%0A%20%20%20%20x_moons%2C%0A%20%20%20%20y_hi%2C%0A%20%20%20%20y_lo%2C%0A%20%20%20%20y_moons%2C%0A%20%20%20%20y_moons_coord%2C%0A)%3A%0A%20%20%20%20def%20draw_moons(ax%2C%20widget)%3A%0A%20%20%20%20%20%20%20%20for%20c%20in%20range(2)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20mask%20%3D%20y_moons%20%3D%3D%20c%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.scatter(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20x_moons%5Bmask%5D%2C%20y_moons_coord%5Bmask%5D%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20alpha%3D0.6%2C%20color%3DCOLORS%5Bc%5D%2C%20s%3D20%2C%20label%3Df%22Class%20%7Bc%7D%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20ax.legend(fontsize%3D8)%0A%20%20%20%20%20%20%20%20ax.set_xlabel(%22X%22)%0A%20%20%20%20%20%20%20%20ax.set_ylabel(%22Y%22)%0A%20%20%20%20%20%20%20%20ax.grid(True%2C%20alpha%3D0.3)%0A%20%20%20%20%20%20%20%20ax.set_title(%22Draw%20selections%20to%20classify%20the%20two%20moons%22)%0A%0A%20%20%20%20ms%20%3D%20ChartMultiSelect.from_callback(%0A%20%20%20%20%20%20%20%20draw_fn%3Ddraw_moons%2C%0A%20%20%20%20%20%20%20%20x_bounds%3D(x_lo%2C%20x_hi)%2C%0A%20%20%20%20%20%20%20%20y_bounds%3D(y_lo%2C%20y_hi)%2C%0A%20%20%20%20%20%20%20%20figsize%3D(6%2C%204)%2C%0A%20%20%20%20%20%20%20%20n_classes%3D2%2C%0A%20%20%20%20%20%20%20%20mode%3D%22lasso%22%2C%0A%20%20%20%20)%0A%20%20%20%20return%20(ms%2C)%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20ms)%3A%0A%20%20%20%20widget%20%3D%20mo.ui.anywidget(ms)%0A%20%20%20%20return%20(widget%2C)%0A%0A%0A%40app.cell%0Adef%20_(widget)%3A%0A%20%20%20%20widget%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20np%2C%20widget%2C%20x_moons%2C%20y_moons%2C%20y_moons_coord)%3A%0A%20%20%20%20_labels%20%3D%20widget.get_labels(x_moons%2C%20y_moons_coord)%0A%20%20%20%20_classified%20%3D%20_labels%20%3E%3D%200%0A%20%20%20%20_msg%20%3D%20%22Draw%20selections%20to%20classify%20points%2C%20then%20check%20accuracy%20here.%22%0A%20%20%20%20if%20_classified.any()%3A%0A%20%20%20%20%20%20%20%20_correct%20%3D%20int(np.sum(_labels%5B_classified%5D%20%3D%3D%20y_moons%5B_classified%5D))%0A%20%20%20%20%20%20%20%20_total%20%3D%20int(_classified.sum())%0A%20%20%20%20%20%20%20%20_acc%20%3D%20_correct%20%2F%20_total%20*%20100%0A%20%20%20%20%20%20%20%20_msg%20%3D%20f%22Accuracy%20on%20selected%20points%3A%20%7B_correct%7D%2F%7B_total%7D%20(%7B_acc%3A.1f%7D%25)%22%0A%20%20%20%20mo.callout(_msg)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(%22%22%22%0A%20%20%20%20%23%23%20Three-Class%20Blobs%0A%0A%20%20%20%20Using%20%60n_classes%3D3%60%20with%20sklearn%20blobs%20%E2%80%94%20try%20to%20separate%20all%20three%20clusters.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(make_blobs%2C%20np)%3A%0A%20%20%20%20X_blobs%2C%20y_blobs%20%3D%20make_blobs(%0A%20%20%20%20%20%20%20%20n_samples%3D300%2C%20centers%3D3%2C%20cluster_std%3D0.8%2C%20random_state%3D7%2C%0A%20%20%20%20)%0A%20%20%20%20x_blobs%20%3D%20X_blobs%5B%3A%2C%200%5D%0A%20%20%20%20y_blobs_coord%20%3D%20X_blobs%5B%3A%2C%201%5D%0A%20%20%20%20pad3%20%3D%201.0%0A%20%20%20%20x3_lo%2C%20x3_hi%20%3D%20float(np.min(x_blobs)%20-%20pad3)%2C%20float(np.max(x_blobs)%20%2B%20pad3)%0A%20%20%20%20y3_lo%2C%20y3_hi%20%3D%20float(np.min(y_blobs_coord)%20-%20pad3)%2C%20float(np.max(y_blobs_coord)%20%2B%20pad3)%0A%20%20%20%20return%20x3_hi%2C%20x3_lo%2C%20x_blobs%2C%20y3_hi%2C%20y3_lo%2C%20y_blobs%2C%20y_blobs_coord%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(%0A%20%20%20%20COLORS%2C%0A%20%20%20%20ChartMultiSelect%2C%0A%20%20%20%20x3_hi%2C%0A%20%20%20%20x3_lo%2C%0A%20%20%20%20x_blobs%2C%0A%20%20%20%20y3_hi%2C%0A%20%20%20%20y3_lo%2C%0A%20%20%20%20y_blobs%2C%0A%20%20%20%20y_blobs_coord%2C%0A)%3A%0A%20%20%20%20def%20draw_blobs(ax%2C%20widget)%3A%0A%20%20%20%20%20%20%20%20for%20c%20in%20range(3)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20mask%20%3D%20y_blobs%20%3D%3D%20c%0A%20%20%20%20%20%20%20%20%20%20%20%20ax.scatter(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20x_blobs%5Bmask%5D%2C%20y_blobs_coord%5Bmask%5D%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20alpha%3D0.6%2C%20color%3DCOLORS%5Bc%5D%2C%20s%3D20%2C%20label%3Df%22Class%20%7Bc%7D%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20ax.legend(fontsize%3D8)%0A%20%20%20%20%20%20%20%20ax.grid(True%2C%20alpha%3D0.3)%0A%20%20%20%20%20%20%20%20ax.set_title(%22Draw%20selections%20to%20classify%20the%20three%20blobs%22)%0A%0A%20%20%20%20ms3%20%3D%20ChartMultiSelect.from_callback(%0A%20%20%20%20%20%20%20%20draw_fn%3Ddraw_blobs%2C%0A%20%20%20%20%20%20%20%20x_bounds%3D(x3_lo%2C%20x3_hi)%2C%0A%20%20%20%20%20%20%20%20y_bounds%3D(y3_lo%2C%20y3_hi)%2C%0A%20%20%20%20%20%20%20%20figsize%3D(6%2C%205)%2C%0A%20%20%20%20%20%20%20%20n_classes%3D3%2C%0A%20%20%20%20%20%20%20%20mode%3D%22box%22%2C%0A%20%20%20%20)%0A%20%20%20%20return%20(ms3%2C)%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20ms3)%3A%0A%20%20%20%20w3%20%3D%20mo.ui.anywidget(ms3)%0A%20%20%20%20return%20(w3%2C)%0A%0A%0A%40app.cell%0Adef%20_(w3)%3A%0A%20%20%20%20w3%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20np%2C%20w3%2C%20x_blobs%2C%20y_blobs%2C%20y_blobs_coord)%3A%0A%20%20%20%20_labels%20%3D%20w3.get_labels(x_blobs%2C%20y_blobs_coord)%0A%20%20%20%20_classified%20%3D%20_labels%20%3E%3D%200%0A%20%20%20%20_msg%20%3D%20%22Draw%20selections%20to%20classify%20points%2C%20then%20check%20accuracy%20here.%22%0A%20%20%20%20if%20_classified.any()%3A%0A%20%20%20%20%20%20%20%20_correct%20%3D%20int(np.sum(_labels%5B_classified%5D%20%3D%3D%20y_blobs%5B_classified%5D))%0A%20%20%20%20%20%20%20%20_total%20%3D%20int(_classified.sum())%0A%20%20%20%20%20%20%20%20_acc%20%3D%20_correct%20%2F%20_total%20*%20100%0A%20%20%20%20%20%20%20%20_msg%20%3D%20f%22Accuracy%20on%20selected%20points%3A%20%7B_correct%7D%2F%7B_total%7D%20(%7B_acc%3A.1f%7D%25)%22%0A%20%20%20%20mo.callout(_msg)%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A