U
    Kvf                     @   s:  d Z ddlZddlZejd dZdZdZdZ	dD ]Z
edd	e
Zd
D ]ZdD ]ZeddeZg Zg Zg Zg ZeeD ]ZejeedZejjeefdZeee
fZeddeZee
D ]Zee eddef< qejje
de Zeeeeee Z e!e eejjed   e!e e!e e!ee"e  qe#eZej#eddZej#eddZej#eddZej#edddf edddf eefddZ$dgdd eeD  dd ee
D  Z%d&e%Z%ej'(ej')e*Z+ej'&e+de	 Z,ej-e,e$de%ddd e	d7 Z	qVqLq4dS )zm
Generate test data sets for lme.

After running this script, run lme_results.R with R
to update the output.
    NiKQ d         )r      g      g      ?)r   r      )g      ?r   )size)Zaxiszgroups,endogc                 C   s   g | ]}d | qS )z
exog_fe_%d .0kr	   r	   U/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/regression/tests/results/generate_lme.py
<listcomp>B   s     r   c                 C   s   g | ]}d | qS )z
exog_re_%dr	   r
   r	   r	   r   r   C   s     ,zlme%02d.csvz%.3f )fmtheader	delimitercomments).__doc__ZnumpynposrandomseedZngroupZn_minZn_maxZdsixprZlinspaceZre_sdpfsigparamsZendogZexog_feZexog_regroupsrangeirandintnnormalZx_fezerosZx_reujredotZexpvalappendZonesZconcatenatedatar   joinpathdirnameabspath__file__Zcur_dirfnameZsavetxtr	   r	   r	   r   <module>   sh   





  
 