B
    (b                 @   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   S/home/dcms/DCMS/lib/python3.7/site-packages/sklearn/model_selection/tests/common.py__init__   s    zOneTimeSplitter.__init__Nc             c   s   x| j D ]
}|V  qW 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 )N)r   )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   numpyr
   Zsklearn.model_selectionr   r   r   r   r   r   <module>   s   