U
    Kvf                     @   sH   d Z ddlmZ dZdZe ZdZdZeZ	dZ
dd	 Zd
d Zdd ZdS )z6Spector and Mazzeo (1980) - Program Effectiveness Data    )utilsZrestructuredtextzMUsed with express permission of the original author, who
retains all rights. a  
http://pages.stern.nyu.edu/~wgreene/Text/econometricanalysis.htm

The raw data was downloaded from Bill Greene's Econometric Analysis web site,
though permission was obtained from the original researcher, Dr. Lee Spector,
Professor of Economics, Ball State University.z^Experimental data on the effectiveness of the personalized
system of instruction (PSI) programaj  ::

    Number of Observations - 32

    Number of Variables - 4

    Variable name definitions::

        Grade - binary variable indicating whether or not a student's grade
                improved.  1 indicates an improvement.
        TUCE  - Test score on economics test
        PSI   - participation in program
        GPA   - Student's grade point average
c                   C   s   t  S )
    Load the Spector dataset and returns a Dataset class instance.

    Returns
    -------
    Dataset
        See DATASET_PROPOSAL.txt for more information.
    )load_pandas r   r   E/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/datasets/spector/data.pyload%   s    	r   c                  C   s   t  } tj| ddS )r      )Z	endog_idx)	_get_dataduZprocess_pandasdatar   r   r   r   1   s    	r   c                  C   s>   t jtddd} t | } | jd d ddddgf } | tS )Nzspector.csvz\s)sep      r      )r
   Zload_csv__file__Zstrip_column_namesZilocZastypefloatr   r   r   r   r	   >   s    
r	   N)__doc__Zstatsmodels.datasetsr   r
   Z__docformat__Z	COPYRIGHTZTITLEZSOURCEZ
DESCRSHORTZ	DESCRLONGZNOTEr   r   r	   r   r   r   r   <module>   s   