U md@sdZddlmZddlmZmZddlZddlm Z ddl Z ddl m Z ddl mZdd l mZdd l mZdd l mZd d lmZmZeddddZddddddddZdS)aModified 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/ )exists)makedirsremoveN)loadmat) get_data_home) _fetch_remote)RemoteFileMetadata) _pkl_filepath) load_descr)check_random_stateBunchzolivettifaces.matz.https://ndownloader.figshare.com/files/5976027Z@b612fb967f2dc77c9c62d3e1266e0c73d5fca46a4b8906c18e454d41af987794)filenameurlZchecksumFT) data_homeshuffle random_statedownload_if_missing return_X_yc CsDt|d}t|st|t|d}t|s|s8tdtdtj|ftt|d}t |d}t ||dj }t j||dd ~n t |}t|}||}||}|d d d d }tddtdD} |rt|}|t|} || }| | } |t|d} td} |r4| | fSt| || | dS)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 `. Parameters ---------- data_home : str, 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 `. download_if_missing : bool, default=True If False, raise a IOError 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 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 )rz olivetti.pkzz1Data not found and `download_if_missing` is Falsez(downloading Olivetti faces from %s to %s)dirname) file_namefaces)compress)@rrr rcSsg|] }|dqS) ).0irrY/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/sklearn/datasets/_olivetti_faces.py sz(fetch_olivetti_faces..rzolivetti_faces.rst)dataZimagestargetZDESCR)rrrr IOErrorprintFACESrrrrTcopyjoblibdumploadnpZfloat32minmaxZreshapeZ transposearrayranger Z permutationlenr r) rrrrrfilepathZmat_pathZmfilerr%orderZfaces_vectorizedZfdescrrrr!fetch_olivetti_faces's<E        r6)__doc__os.pathrosrrnumpyr.Zscipy.iorr+r_baserr r r utilsr rr(r6rrrr!s,