Exemplo n.º 1
0
Query the number of mice at different timepoints of the pipeline

@author: Anne Urai, Guido Meijer, Miles Wells, 16 Jan 2020
Updated 22 April 2020, Anne Urai
"""

from paper_behavior_functions import query_subjects, CUTOFF_DATE
from ibl_pipeline import subject, acquisition, reference
from ibl_pipeline.analyses import behavior as behavior_analysis

# =========================
# 1. Query all mice on brainwide map project which began training before the paper's cutoff date
# =========================

all_mice = (subject.Subject * subject.SubjectLab * reference.Lab *
            subject.SubjectProject()
            & 'subject_project = "ibl_neuropixel_brainwide_01"').aggr(
                acquisition.Session,
                first_session='min(date(session_start_time))')

# Filter mice that started training after the paper's cutoff date
all_mice = all_mice.aggr(acquisition.Session,
                         first_session='min(date(session_start_time))')
all_mice = (all_mice & 'first_session < "%s"' % CUTOFF_DATE)

print('1. Total # of mice in brainwide project: %d' % len(all_mice))

# ==================================================
# Exclude mice that are still in training at the date of cut-off, meaning they have not yet
# reached any learned criteria
# ==================================================
import seaborn as sns
from ibl_pipeline import subject, reference, acquisition
from paper_behavior_functions import (seaborn_style, institution_map,
                                      group_colors, figpath, CUTOFF_DATE,
                                      FIGURE_HEIGHT, FIGURE_WIDTH, QUERY)
from ibl_pipeline.analyses import behavior as behavior_analysis
from lifelines import KaplanMeierFitter

# Settings
fig_path = figpath()
seaborn_style()

if QUERY is True:
    # Query all mice
    all_mice = (subject.Subject * subject.SubjectLab * reference.Lab
                * subject.SubjectProject() & 'subject_project = "ibl_neuropixel_brainwide_01"')
    mice_started_training = (all_mice & (acquisition.Session() & 'task_protocol LIKE "%training%"'))
    still_training = all_mice.aggr(behavior_analysis.SessionTrainingStatus,
                                   session_start_time='max(session_start_time)') \
                                        * behavior_analysis.SessionTrainingStatus - subject.Death \
                                        & 'training_status = "in_training"' \
                                        & 'session_start_time > "%s"' % CUTOFF_DATE
    use_subjects = mice_started_training - still_training
    
    # Get training status and training time in number of sessions and trials
    ses = (use_subjects
           * behavior_analysis.SessionTrainingStatus
           * behavior_analysis.PsychResults).proj(
                   'subject_nickname', 'training_status', 'n_trials_stim', 'institution_short').fetch(
                                                                       format='frame').reset_index()
    ses['n_trials'] = [sum(i) for i in ses['n_trials_stim']]