>        Args:</span>
<span class="sd">            images (numpy.array/str): array with shape (n_images, dim1, dim2 , channel_size), or a string with name of keras-dataset (cifar10, fashion_mnsit)</span>
<span class="sd">            labels (numpy.array): labels of images, array with shape (n_images) where each element is an integer from 0 to number of classes</span>
<span class="sd">        &quot;&quot;&quot;</span>
        <span class="k">if</span> <span class="nb">type</span><span class="p">(</span><span 