U
    ÃmœdÑ  ã                   @   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é    )ÚutilsÚ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   úZ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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   Ú__docformat__Z	COPYRIGHTZTITLEZSOURCEZ
DESCRSHORTZ	DESCRLONGÚNOTEr   r   r
   r   r   r   r   Ú<module>   s   