U
    Ãmœd»	  ã                   @   sH   d Z ddlmZ dZdZdZdZdZdZ	d	Z
d
d„ Zdd„ Zdd„ ZdS )z Fair's Extramarital Affairs Dataé    )ÚutilsÚrestructuredtextz'Included with permission of the author.zAffairs datasetz¶
Fair, Ray. 1978. "A Theory of Extramarital Affairs," `Journal of Political
Economy`, February, 45-61.

The data is available at http://fairmodel.econ.yale.edu/rayfair/pdf/2011b.htm
zExtramarital affair data.zÖExtramarital affair data used to explain the allocation
of an individual's time among work, time spent with a spouse, and time
spent with a paramour. The data is used as an example of regression
with censored data.aL  ::

    Number of observations: 6366
    Number of variables: 9
    Variable name definitions:

        rate_marriage   : How rate marriage, 1 = very poor, 2 = poor, 3 = fair,
                        4 = good, 5 = very good
        age             : Age
        yrs_married     : No. years married. Interval approximations. See
                        original paper for detailed explanation.
        children        : No. children
        religious       : How relgious, 1 = not, 2 = mildly, 3 = fairly,
                        4 = strongly
        educ            : Level of education, 9 = grade school, 12 = high
                        school, 14 = some college, 16 = college graduate,
                        17 = some graduate school, 20 = advanced degree
        occupation      : 1 = student, 2 = farming, agriculture; semi-skilled,
                        or unskilled worker; 3 = white-colloar; 4 = teacher
                        counselor social worker, nurse; artist, writers;
                        technician, skilled worker, 5 = managerial,
                        administrative, business, 6 = professional with
                        advanced degree
        occupation_husb : Husband's occupation. Same as occupation.
        affairs         : measure of time spent in extramarital affairs

    See the original paper for more details.
c                   C   s   t ƒ S )z˜
    Load the data and return a Dataset class instance.

    Returns
    -------
    Dataset
        See DATASET_PROPOSAL.txt for more information.
    )Úload_pandas© r   r   úW/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/datasets/fair/data.pyÚload5   s    	r   c                  C   s   t ƒ } tj| dd d�S )Né   )Z	endog_idxZexog_idx)Ú	_get_dataÚduZprocess_pandas)Údatar   r   r   r   A   s    r   c                   C   s   t jtddd�S )Nzfair.csvT)Zconvert_float)r
   Zload_csvÚ__file__r   r   r   r   r	   F   s    r	   N)Ú__doc__Zstatsmodels.datasetsr   r
   Ú__docformat__Z	COPYRIGHTZTITLEZSOURCEZ
DESCRSHORTZ	DESCRLONGÚNOTEr   r   r	   r   r   r   r   Ú<module>   s   