U
    2d                     @   s*   d Z ddlZddlmZ G dd dZdS )z/
Common utilities for testing model selection.
    N)KFoldc                   @   s.   e Zd ZdZdddZdddZdd	d
ZdS )OneTimeSplitterz0A wrapper to make KFold single entry cv iterator   c   c                 C   s,   || _ || _tt|dt|| _d S )Nn_splits)r   	n_samplesiterr   splitnpZonesindices)selfr   r    r   H/tmp/pip-unpacked-wheel-zrfo1fqw/sklearn/model_selection/tests/common.py__init__   s    zOneTimeSplitter.__init__Nc                 c   s   | j D ]
}|V  qdS )zSplit can be called only onceN)r   )r   Xygroupsindexr   r   r   r
      s    
zOneTimeSplitter.splitc                 C   s   | j S )Nr   )r   r   r   r   r   r   r   get_n_splits   s    zOneTimeSplitter.get_n_splits)r   r   )NNN)NNN)__name__
__module____qualname____doc__r   r
   r   r   r   r   r   r   
   s   

r   )r   Znumpyr   Zsklearn.model_selectionr   r   r   r   r   r   <module>   s   