U f»{j²ã@s*ddlZddlmZdd„Zd dd„ZdS) éNé)Ú _check_optioncCs6|jj|jjkrtdƒ‚t |j|jk¡r2tdƒ‚dS)zCheck that stcs are compatible.z$Data in stcs must have the same sizezTimes of two stcs must match.N)ÚdataÚshapeÚ ValueErrorÚnpÚallÚtimes)Ústc1Ústc2©r úU/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/simulation/metrics.pyÚ _check_stcsrÚrmscCs†td|ddgƒt||ƒ|j|j}}|dkrJt t ||d¡¡}n8|dkr‚dt | ¡| ¡¡tj  |¡tj  |¡}|S)a[Calculate matrix similarities. Parameters ---------- stc1 : SourceEstimate First source estimate for comparison. stc2 : SourceEstimate Second source estimate for comparison. metric : str Metric to calculate, 'rms' or 'cosine'. Returns ------- score : float | array Calculated metric. Notes ----- Metric calculation has multiple options: * rms: Root mean square of difference between stc data matrices. * cosine: Normalized correlation of all elements in stc data matrices. .. versionadded:: 0.10.0 ÚmetricrZcosinergð?) rrrrÚsqrtZmeanÚdotÚflattenZlinalgZnorm)r r rZdata1Zdata2Zscorer r r Úsource_estimate_quantifications ÿr)r)ÚnumpyrÚutilsrrrr r r r Ús