a
    �ù»js  ã                   @   s  d Z ddlmZmZ ddlmZmZmZ ddlm	Z	 ddl
Z
ddlZddlmZ ddlmZmZ dd	lmZmZ d
dlmZ d
dlmZmZmZmZ edddd�Zeeedgdgdgdgdgeed
ddd�geedddd�gdœdd�ddddddddœdd„ƒZdS )a  Modified Olivetti faces dataset.

The original database was available from (now defunct)

    https://www.cl.cam.ac.uk/research/dtg/attarchive/facedatabase.html

The version retrieved here comes in MATLAB format from the personal
web page of Sam Roweis:

    https://cs.nyu.edu/~roweis/
é    )ÚIntegralÚReal)ÚPathLikeÚmakedirsÚremove)ÚexistsN)Úloadmaté   )ÚBunchÚcheck_random_state)ÚIntervalÚvalidate_paramsé   )Úget_data_home)ÚRemoteFileMetadataÚ_fetch_remoteÚ_pkl_filepathÚ
load_descrzolivettifaces.matz.https://ndownloader.figshare.com/files/5976027Z@b612fb967f2dc77c9c62d3e1266e0c73d5fca46a4b8906c18e454d41af987794)ÚfilenameÚurlZchecksumÚbooleanÚrandom_stateÚleft)Úclosedg        Zneither)Ú	data_homeÚshuffler   Údownload_if_missingÚ
return_X_yÚ	n_retriesÚdelayT)Zprefer_skip_nested_validationFé   g      ð?c                 C   sH  t | d�} t| ƒst| ƒ t| dƒ}t|ƒsŽ|s8tdƒ‚tdtj| f ƒ tt| ||d�}t	|d�}	t
|ƒ |	d j ¡ }
tj|
|dd	� ~	n
t |¡}
t |
¡}
|
|
 ¡  }
|
|
 ¡  }
|
 d
¡ ddd¡}
t dd„ tdƒD ƒ¡}|�rt|ƒ}| t|
ƒ¡}|
| }
|| }|
 t|
ƒd¡}tdƒ}|�r8||fS t||
||d�S )a  Load the Olivetti faces data-set from AT&T (classification).

    Download it if necessary.

    =================   =====================
    Classes                                40
    Samples total                         400
    Dimensionality                       4096
    Features            real, between 0 and 1
    =================   =====================

    Read more in the :ref:`User Guide <olivetti_faces_dataset>`.

    Parameters
    ----------
    data_home : str or path-like, default=None
        Specify another download and cache folder for the datasets. By default
        all scikit-learn data is stored in '~/scikit_learn_data' subfolders.

    shuffle : bool, default=False
        If True the order of the dataset is shuffled to avoid having
        images of the same person grouped.

    random_state : int, RandomState instance or None, default=0
        Determines random number generation for dataset shuffling. Pass an int
        for reproducible output across multiple function calls.
        See :term:`Glossary <random_state>`.

    download_if_missing : bool, default=True
        If False, raise an OSError if the data is not locally available
        instead of trying to download the data from the source site.

    return_X_y : bool, default=False
        If True, returns `(data, target)` instead of a `Bunch` object. See
        below for more information about the `data` and `target` object.

        .. versionadded:: 0.22

    n_retries : int, default=3
        Number of retries when HTTP errors are encountered.

        .. versionadded:: 1.5

    delay : float, default=1.0
        Number of seconds between retries.

        .. versionadded:: 1.5

    Returns
    -------
    data : :class:`~sklearn.utils.Bunch`
        Dictionary-like object, with the following attributes.

        data: ndarray, shape (400, 4096)
            Each row corresponds to a ravelled
            face image of original size 64 x 64 pixels.
        images : ndarray, shape (400, 64, 64)
            Each row is a face image
            corresponding to one of the 40 subjects of the dataset.
        target : ndarray, shape (400,)
            Labels associated to each face image.
            Those labels are ranging from 0-39 and correspond to the
            Subject IDs.
        DESCR : str
            Description of the modified Olivetti Faces Dataset.

    (data, target) : tuple if `return_X_y=True`
        Tuple with the `data` and `target` objects described above.

        .. versionadded:: 0.22

    Examples
    --------
    >>> from sklearn.datasets import fetch_olivetti_faces
    >>> olivetti_faces = fetch_olivetti_faces()
    >>> olivetti_faces.data.shape
    (400, 4096)
    >>> olivetti_faces.target.shape
    (400,)
    >>> olivetti_faces.images.shape
    (400, 64, 64)
    )r   zolivetti.pkzz1Data not found and `download_if_missing` is Falsez(downloading Olivetti faces from %s to %s)Údirnamer   r   )Ú	file_nameÚfacesé   )Úcompress)é�  é@   r'   r   r	   r   c                 S   s   g | ]}|d  ‘qS )é
   © )Ú.0Úir)   r)   úFE:\soccer\ml_env\Lib\site-packages\sklearn/datasets/_olivetti_faces.pyÚ
<listcomp>«   ó    z(fetch_olivetti_faces.<locals>.<listcomp>r&   éÿÿÿÿzolivetti_faces.rst)ÚdataZimagesÚtargetZDESCR)r   r   r   r   ÚOSErrorÚprintÚFACESr   r   r   r   ÚTÚcopyÚjoblibÚdumpÚloadÚnpZfloat32ÚminÚmaxZreshapeZ	transposeÚarrayÚranger   ZpermutationÚlenr   r
   )r   r   r   r   r   r   r   ÚfilepathZmat_pathZmfiler#   r1   ÚorderZfaces_vectorizedZfdescrr)   r)   r,   Úfetch_olivetti_faces&   s@    h

ÿ


rB   )Ú__doc__Únumbersr   r   Úosr   r   r   Úos.pathr   r7   Únumpyr:   Zscipy.ior   Úutilsr
   r   Zutils._param_validationr   r   Ú r   Z_baser   r   r   r   r4   ÚstrrB   r)   r)   r)   r,   Ú<module>   sD   ýù	öø