.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/datasets/plot_random_multilabel_dataset.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_auto_examples_datasets_plot_random_multilabel_dataset.py>`
        to download the full example code. or to run this example in your browser via Binder

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_datasets_plot_random_multilabel_dataset.py:


==============================================
绘制随机生成的多标签数据集
==============================================

这展示了 :func:`~sklearn.datasets.make_multilabel_classification` 数据集生成器。每个样本由两个特征的计数（最多50个）组成，这些特征在两个类别中有不同的分布。

点的标签如下，其中 Y 表示类别存在：

    =====  =====  =====  ======
      1      2      3    颜色
    =====  =====  =====  ======
      Y      N      N    红色
      N      Y      N    蓝色
      N      N      Y    黄色
      Y      Y      N    紫色
      Y      N      Y    橙色
      Y      Y      N    绿色
      Y      Y      Y    棕色
    =====  =====  =====  ======

星号标记了每个类别的预期样本；其大小反映了选择该类别标签的概率。

左右两个示例突出了 ``n_labels`` 参数：右图中的更多样本具有2或3个标签。

请注意，这个二维示例非常退化：通常特征的数量会远大于“文档长度”，而在这里我们有比词汇量大得多的文档。同样，当 ``n_classes > n_features`` 时，某个特征区分特定类别的可能性要小得多。

.. GENERATED FROM PYTHON SOURCE LINES 29-98



.. image-sg:: /auto_examples/datasets/images/sphx_glr_plot_random_multilabel_dataset_001.png
   :alt: n_labels=1, length=50, n_labels=3, length=50
   :srcset: /auto_examples/datasets/images/sphx_glr_plot_random_multilabel_dataset_001.png
   :class: sphx-glr-single-img


.. rst-class:: sphx-glr-script-out

 .. code-block:: none

    The data was generated from (random_state=669):
    Class   P(C)    P(w0|C) P(w1|C)
    red     0.36    0.16    0.84
    blue    0.41    0.30    0.70
    yellow  0.23    0.42    0.58






|

.. code-block:: Python


    import matplotlib.pyplot as plt
    import numpy as np

    from sklearn.datasets import make_multilabel_classification as make_ml_clf

    COLORS = np.array(
        [
            "!",
            "#FF3333",  # red
            "#0198E1",  # blue
            "#BF5FFF",  # purple
            "#FCD116",  # yellow
            "#FF7216",  # orange
            "#4DBD33",  # green
            "#87421F",  # brown
        ]
    )

    # 使用相同的随机种子多次调用 make_multilabel_classification 以确保相同的分布
    RANDOM_SEED = np.random.randint(2**10)


    def plot_2d(ax, n_labels=1, n_classes=3, length=50):
        X, Y, p_c, p_w_c = make_ml_clf(
            n_samples=150,
            n_features=2,
            n_classes=n_classes,
            n_labels=n_labels,
            length=length,
            allow_unlabeled=False,
            return_distributions=True,
            random_state=RANDOM_SEED,
        )

        ax.scatter(
            X[:, 0], X[:, 1], color=COLORS.take((Y * [1, 2, 4]).sum(axis=1)), marker="."
        )
        ax.scatter(
            p_w_c[0] * length,
            p_w_c[1] * length,
            marker="*",
            linewidth=0.5,
            edgecolor="black",
            s=20 + 1500 * p_c**2,
            color=COLORS.take([1, 2, 4]),
        )
        ax.set_xlabel("Feature 0 count")
        return p_c, p_w_c


    _, (ax1, ax2) = plt.subplots(1, 2, sharex="row", sharey="row", figsize=(8, 4))
    plt.subplots_adjust(bottom=0.15)

    p_c, p_w_c = plot_2d(ax1, n_labels=1)
    ax1.set_title("n_labels=1, length=50")
    ax1.set_ylabel("Feature 1 count")

    plot_2d(ax2, n_labels=3)
    ax2.set_title("n_labels=3, length=50")
    ax2.set_xlim(left=0, auto=True)
    ax2.set_ylim(bottom=0, auto=True)

    plt.show()

    print("The data was generated from (random_state=%d):" % RANDOM_SEED)
    print("Class", "P(C)", "P(w0|C)", "P(w1|C)", sep="\t")
    for k, p, p_w in zip(["red", "blue", "yellow"], p_c, p_w_c.T):
        print("%s\t%0.2f\t%0.2f\t%0.2f" % (k, p, p_w[0], p_w[1]))


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 0.057 seconds)


.. _sphx_glr_download_auto_examples_datasets_plot_random_multilabel_dataset.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: binder-badge

      .. image:: images/binder_badge_logo.svg
        :target: https://mybinder.org/v2/gh/scikit-learn/scikit-learn/main?urlpath=lab/tree/notebooks/auto_examples/datasets/plot_random_multilabel_dataset.ipynb
        :alt: Launch binder
        :width: 150 px

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: plot_random_multilabel_dataset.ipynb <plot_random_multilabel_dataset.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: plot_random_multilabel_dataset.py <plot_random_multilabel_dataset.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: plot_random_multilabel_dataset.zip <plot_random_multilabel_dataset.zip>`


.. include:: plot_random_multilabel_dataset.recommendations


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_