.. DO NOT EDIT. .. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY. .. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE: .. "generated/sphinx_gallery_examples/pool/plot-UncertaintySampling-Density-weighted_Uncertainty_Sampling.py" .. LINE NUMBERS ARE GIVEN BELOW. .. only:: html .. note:: :class: sphx-glr-download-link-note :ref:`Go to the end ` to download the full example code. .. rst-class:: sphx-glr-example-title .. _sphx_glr_generated_sphinx_gallery_examples_pool_plot-UncertaintySampling-Density-weighted_Uncertainty_Sampling.py: Density-weighted Uncertainty Sampling ===================================== .. GENERATED FROM PYTHON SOURCE LINES 10-23 .. note:: The generated animation can be found at the bottom of the page. | **Google Colab Note**: If the notebook fails to run after installing the needed packages, try to restart the runtime (Ctrl + M) under Runtime -> Restart session. .. image:: https://colab.research.google.com/assets/colab-badge.svg :target: https://colab.research.google.com/github/scikit-activeml/scikit-activeml.github.io/blob/gh-pages/latest/generated/sphinx_gallery_notebooks//pool/plot-UncertaintySampling-Density-weighted_Uncertainty_Sampling.ipynb | **Notebook Dependencies** | Uncomment the following cell to install all dependencies for this tutorial. .. GENERATED FROM PYTHON SOURCE LINES 23-26 .. code-block:: Python # !pip install scikit-activeml .. GENERATED FROM PYTHON SOURCE LINES 27-31 .. raw:: html
.. GENERATED FROM PYTHON SOURCE LINES 33-121 .. code-block:: Python import numpy as np from matplotlib import pyplot as plt, animation from sklearn.datasets import make_blobs from skactiveml.utils import MISSING_LABEL, labeled_indices, unlabeled_indices from skactiveml.visualization import plot_utilities, plot_decision_boundary from sklearn.linear_model import LogisticRegression from sklearn.mixture import GaussianMixture from skactiveml.classifier import SklearnClassifier from skactiveml.pool import UncertaintySampling random_state = np.random.RandomState(0) # Build a dataset. X, y_true = make_blobs( n_samples=200, n_features=2, centers=[[0, 1], [-3, 0.5], [-1, -1], [2, 1], [1, -0.5]], cluster_std=0.7, random_state=random_state, ) y_true = y_true % 2 y = np.full(shape=y_true.shape, fill_value=MISSING_LABEL) # Initialise the classifier. clf = SklearnClassifier(LogisticRegression(), classes=np.unique(y_true)) # Initialise the query strategy. qs = UncertaintySampling(method='least_confident', random_state=random_state) gmm = GaussianMixture(init_params='kmeans', n_components=5) gmm.fit(X) density = np.exp(gmm.score_samples(X)) # Preparation for plotting. fig, ax = plt.subplots() feature_bound = [[min(X[:, 0]), min(X[:, 1])], [max(X[:, 0]), max(X[:, 1])]] artists = [] # Active learning cycle: n_cycles = 20 for c in range(n_cycles): # Fit the classifier with current labels. clf.fit(X, y) # Query the next sample(s). query_idx = qs.query(X=X, y=y, clf=clf, utility_weight=density) # Capture the current plot state. coll_old = list(ax.collections) title = ax.text( 0.5, 1.05, f"Decision boundary after acquiring {c} labels", size=plt.rcParams["axes.titlesize"], ha="center", transform=ax.transAxes, ) # Update plot with utility values, samples, and decision boundary. X_labeled = X[labeled_indices(y)] ax = plot_utilities( qs, X=X, y=y, clf=clf, utility_weight=density, candidates=unlabeled_indices(y), replace_nan=None, res=25, feature_bound=feature_bound, ax=ax, ) ax.scatter( X[:, 0], X[:, 1], c=y_true, cmap="coolwarm", marker=".", zorder=2 ) ax.scatter( X_labeled[:, 0], X_labeled[:, 1], c="grey", alpha=0.8, marker=".", s=300, ) ax = plot_decision_boundary(clf, feature_bound, ax=ax) coll_new = list(ax.collections) coll_new.append(title) artists.append([x for x in coll_new if x not in coll_old]) # Update labels based on query. y[query_idx] = y_true[query_idx] ani = animation.ArtistAnimation(fig, artists, interval=1000, blit=True) .. container:: sphx-glr-animation .. raw:: html
.. GENERATED FROM PYTHON SOURCE LINES 122-123 .. image:: ../../examples/pool_classification_legend.png .. GENERATED FROM PYTHON SOURCE LINES 125-130 .. rubric:: References: The implementation of this strategy is based on :footcite:t:`tang2002active`. .. footbibliography:: .. rst-class:: sphx-glr-timing **Total running time of the script:** (0 minutes 5.677 seconds) .. _sphx_glr_download_generated_sphinx_gallery_examples_pool_plot-UncertaintySampling-Density-weighted_Uncertainty_Sampling.py: .. only:: html .. container:: sphx-glr-footer sphx-glr-footer-example .. container:: sphx-glr-download sphx-glr-download-jupyter :download:`Download Jupyter notebook: plot-UncertaintySampling-Density-weighted_Uncertainty_Sampling.ipynb ` .. container:: sphx-glr-download sphx-glr-download-python :download:`Download Python source code: plot-UncertaintySampling-Density-weighted_Uncertainty_Sampling.py ` .. container:: sphx-glr-download sphx-glr-download-zip :download:`Download zipped: plot-UncertaintySampling-Density-weighted_Uncertainty_Sampling.zip ` .. only:: html .. rst-class:: sphx-glr-signature `Gallery generated by Sphinx-Gallery `_