ag_tallies[j, :] = trial['ag_tallies']
fd_tallies[j, :] = trial['fd_tallies']
samples = 5.0
van_tallies = van_tallies / samples
van_tallies[:, 2] = 1.0 - van_tallies[:, 0] - van_tallies[:, 1]
rand_tallies = rand_tallies / samples
rand_tallies[:, 2] = 1.0 - rand_tallies[:, 0] - rand_tallies[:, 1]
es_tallies = es_tallies / samples
es_tallies[:, 2] = 1.0 - es_tallies[:, 0] - es_tallies[:, 1]
ag_tallies = ag_tallies / samples
ag_tallies[:, 2] = 1.0 - ag_tallies[:, 0] - ag_tallies[:, 1]
fd_tallies = fd_tallies / samples
fd_tallies[:, 2] = 1.0 - fd_tallies[:, 0] - fd_tallies[:, 1]
van_means, van_sems = statistics.mean_sem(van_tallies)
rand_means, rand_sems = statistics.mean_sem(rand_tallies)
es_means, es_sems = statistics.mean_sem(es_tallies)
ag_means, ag_sems = statistics.mean_sem(ag_tallies)
fd_means, fd_sems = statistics.mean_sem(fd_tallies)
print "Van: " + str(np.mean(van_tallies, axis=0))
print "Early: " + str(np.mean(es_tallies, axis=0))
print "Rand: " + str(np.mean(rand_tallies, axis=0))
print "AG: " + str(np.mean(ag_tallies, axis=0))
print "FD: " + str(np.mean(fd_tallies, axis=0))
print np.sum(fd_means), np.sum(ag_means), np.sum(es_means), np.sum(
van_means), np.sum(rand_means)
plt.style.use('ggplot')
rec_scores = pickle.load(open(data_dir + 'multiple_trials/rec_scores.pkl', 'r')) rec_cutoffs = pickle.load(open(data_dir + 'multiple_trials/rec_cutoffs.pkl', 'r')) rec_mo_scores = pickle.load(open(data_dir + 'multiple_trials/rec_mo_scores.pkl', 'r')) rec_mo_cutoffs = pickle.load(open(data_dir + 'multiple_trials/rec_mo_cutoffs.pkl', 'r')) print "Averaging over " + str(len(all_trials)) + " trials" rec_scores = rec_scores[:] rec_cutoffs = rec_cutoffs[:] rec_mo_scores = rec_mo_scores[:] rec_mo_cutoffs = rec_mo_cutoffs[:] els = 150 data = (np.array(rec_scores) / np.array(rec_cutoffs))[-100: -50] data = np.clip(data, 0, 1) mean, sem = statistics.mean_sem(data) data_mo = (np.array(rec_mo_scores) / np.array(rec_mo_cutoffs))[-100:-50] data_mo = np.clip(data_mo, 0, 1) mean, sem = statistics.mean_sem(data_mo) datas = [data[:, :]] datas_mo = [data_mo[:, :]] labels = ['DFO'] colors = ['blue'] mean = np.mean(data[:, :els], axis=0) std = scipy.stats.sem(data[:, :els], axis=0) mean_mo = np.mean(data_mo[:, :els], axis=0) std_mo = scipy.stats.sem(data_mo[:, :els], axis=0) print mean.shape x = np.arange(mean.shape[0])
es_samples.append(trial['es_tallies'])
van_tallies.append(np.mean(van_samples, axis=0))
rand_tallies.append(np.mean(rand_samples, axis=0))
es_tallies.append(np.mean(es_samples, axis=0))
samples = 1.0
van_tallies = np.array(van_tallies) / samples
rand_tallies = np.array(rand_tallies) / samples
es_tallies = np.array(es_tallies) / samples
van_tallies[:, -1] = 1 - van_tallies[:, 0] - van_tallies[:, 1]
rand_tallies[:, -1] = 1 - rand_tallies[:, 0] - rand_tallies[:, 1]
es_tallies[:, -1] = 1 - es_tallies[:, 0] - es_tallies[:, 1]
van_means, van_sems = statistics.mean_sem(van_tallies)
rand_means, rand_sems = statistics.mean_sem(rand_tallies)
es_means, es_sems = statistics.mean_sem(es_tallies)
# van_means, van_sems = van_means[::-1], van_sems[::-1]
# rand_means, rand_sems = rand_means[::-1], rand_sems[::-1]
# es_means, es_sems = es_means[::-1], es_sems[::-1]
all_van_means.append(van_means)
all_van_sems.append(van_sems)
all_rand_means.append(rand_means)
all_rand_sems.append(rand_sems)
all_es_means.append(es_means)
all_es_sems.append(es_sems)