U
    Kvf	                     @   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    )utilsZ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.zExtramarital 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   B/tmp/pip-unpacked-wheel-2v6byqio/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	   Z__docformat__Z	COPYRIGHTZTITLEZSOURCEZ
DESCRSHORTZ	DESCRLONGZNOTEr   r   r   r   r   r   r   <module>   s   