def create_roc_report(file_prefix,
num_groups,
contrast_range,
num_trials,
reports_dir,
regenerate_plot=True):
num_extra_trials = 10
roc_report = Struct()
roc_report.auc = get_auc(file_prefix, contrast_range, num_trials,
num_extra_trials, num_groups)
roc_report.auc_single_option = []
roc_url = 'img/roc.png'
fname = os.path.join(reports_dir, roc_url)
roc_report.roc_url = roc_url
if regenerate_plot or not os.path.exists(fname):
fig = plt.figure()
for i in range(num_groups):
roc = get_roc_single_option(file_prefix, contrast_range,
num_trials, num_extra_trials, i)
plt.plot(roc[:, 0], roc[:, 1], 'x-', label='option %d' % i)
roc_report.auc_single_option.append(
get_auc_single_option(file_prefix, contrast_range, num_trials,
num_extra_trials, i))
plt.plot([0, 1], [0, 1], '--')
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
save_to_png(fig, fname)
save_to_eps(fig, os.path.join(reports_dir, 'img/roc.eps'))
plt.close()
return roc_report
def plot_roc(data_dir, num_groups, trial_duration, num_trials, p_b_e,p_x_e,p_e_e,p_e_i,p_i_i,p_i_e,num_extra_trials=10):
file_desc='wta.groups.%d.duration.%0.3f.p_b_e.%0.3f.p_x_e.%0.3f.p_e_e.%0.3f.p_e_i.%0.3f.p_i_i.%0.3f.p_i_e.%0.3f' %\
(num_groups, trial_duration, p_b_e, p_x_e, p_e_e, p_e_i, p_i_i, p_i_e)
file_prefix=os.path.join(data_dir,file_desc)
fig=plt.figure()
for i in range(num_groups):
roc=get_roc_single_option(file_prefix, num_trials, num_extra_trials, i)
plt.plot(roc[:,0],roc[:,1],'x-',label='option %d' % i)
plt.plot([0,1],[0,1],'--')
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.show()
def create_roc_report(file_prefix, num_groups, contrast_range, num_trials, reports_dir, regenerate_plot=True):
num_extra_trials=10
roc_report=Struct()
roc_report.auc=get_auc(file_prefix, contrast_range, num_trials, num_extra_trials, num_groups)
roc_report.auc_single_option=[]
roc_url = 'img/roc.png'
fname=os.path.join(reports_dir, roc_url)
roc_report.roc_url=roc_url
if regenerate_plot or not os.path.exists(fname):
fig=plt.figure()
for i in range(num_groups):
roc=get_roc_single_option(file_prefix, contrast_range, num_trials, num_extra_trials, i)
plt.plot(roc[:,0],roc[:,1],'x-',label='option %d' % i)
roc_report.auc_single_option.append(get_auc_single_option(file_prefix, contrast_range, num_trials, num_extra_trials, i))
plt.plot([0,1],[0,1],'--')
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
save_to_png(fig, fname)
save_to_eps(fig, os.path.join(reports_dir, 'img/roc.eps'))
plt.close()
return roc_report
def plot_roc(data_dir,
num_groups,
trial_duration,
num_trials,
p_b_e,
p_x_e,
p_e_e,
p_e_i,
p_i_i,
p_i_e,
num_extra_trials=10):
file_desc='wta.groups.%d.duration.%0.3f.p_b_e.%0.3f.p_x_e.%0.3f.p_e_e.%0.3f.p_e_i.%0.3f.p_i_i.%0.3f.p_i_e.%0.3f' %\
(num_groups, trial_duration, p_b_e, p_x_e, p_e_e, p_e_i, p_i_i, p_i_e)
file_prefix = os.path.join(data_dir, file_desc)
fig = plt.figure()
for i in range(num_groups):
roc = get_roc_single_option(file_prefix, num_trials, num_extra_trials,
i)
plt.plot(roc[:, 0], roc[:, 1], 'x-', label='option %d' % i)
plt.plot([0, 1], [0, 1], '--')
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.show()
