U f{j[ @sVddlZddlmZddlmZddlmZddlm Z m Z m Z m Z e d d d Z dS) N) BaseEpochs) _picks_to_idx)BaseRaw)_check_preload_validate_type _check_optionverboseeogTcCst|dt|ttfddt|j|dd}t|j|}t||rPt d|r\| n|}|j d|ddf}|tj |d d d }|j d kr|d ddt|d }t||j}t|t|f} |dkrt| }d } n"d} tj|td}td|j| ft|D]\} } |j d| ddf} | j}| rv| tj | d d d }|d d }tj|t||jjd|| <| || ||8} q ||fS)aRemove artifacts using regression based on reference channels. Parameters ---------- inst : instance of Epochs | Raw The instance to process. %(picks_good_data)s picks_artifact : array-like | str Channel picks to use as predictor/explanatory variables capturing the artifact of interest (default is "eog"). betas : ndarray, shape (n_picks, n_picks_ref) | None The regression coefficients to use. If None (default), they will be estimated from the data. copy : bool If True (default), copy the instance before modifying it. %(verbose)s Returns ------- inst : instance of Epochs | Raw The processed data. betas : ndarray, shape (n_picks, n_picks_ref) The betas used during regression. Notes ----- To implement the method outlined in :footcite:`GrattonEtAl1983`, remove the evoked response from epochs before estimating the regression coefficients, then apply those regression coefficients to the original data in two calls like (here for a single-condition ``epochs`` only): >>> epochs_no_ave = epochs.copy().subtract_evoked() # doctest:+SKIP >>> _, betas = mne.preprocessing.regress(epochs_no_ave) # doctest:+SKIP >>> epochs_clean, _ = mne.preprocessing.regress(epochs, betas=betas) # doctest:+SKIP References ---------- .. footbibliography:: Zregressinstz Epochs or Rawdata)nonez+picks_artifact cannot be contained in picks.NT)ZkeepdimsrrF)Zdtypez betas.shape)rrrrrinfonpZin1dany ValueErrorcopy_dataZmeanndimZ transposeZreshapelendotTemptyZasarrayfloatrshape enumerateZlinalgZsolve)r ZpicksZpicks_artifactZbetasrr Z artifact_dataZref_dataZcovZ betas_shapeestimatepipickZ this_dataZ orig_shapeZcov_datar"Y/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/_regress.pyregress_artifact s8+     "r$)Nr NTN)numpyrZepochsrZio.pickrZio.baserutilsrrrr r$r"r"r"r#s