def get_final_risk(trace):
col_dict = get_column_dict( trace.db.analyze, 'i', 'salad_risk.tst', 'salad_risk.val', 'argmin_risk.tst', 'argmin_risk.val' )
idx = np.argmax(col_dict.pop('i'))
risk_dict = {}
for key, col in col_dict.iteritems():
risk_dict[key]= col[idx]
return risk_dict
def get_final_predictions(trace):
prediction_dict = {}
col_dict = get_column_dict( trace.db.predict, 'i', 'salad_predict.tst', 'salad_predict.val', 'argmin_predict.val', 'argmin_predict.tst' )
idx = np.argmax(col_dict.pop('i'))
for key, col in col_dict.iteritems():
prediction_dict[key]= col[idx]
return prediction_dict
def get_final_risk(trace):
col_dict = get_column_dict(trace.db.analyze, 'i', 'salad_risk.tst',
'salad_risk.val', 'argmin_risk.tst',
'argmin_risk.val')
idx = np.argmax(col_dict.pop('i'))
risk_dict = {}
for key, col in col_dict.iteritems():
risk_dict[key] = col[idx]
return risk_dict
def get_final_predictions(trace):
prediction_dict = {}
col_dict = get_column_dict(trace.db.predict, 'i', 'salad_predict.tst',
'salad_predict.val', 'argmin_predict.val',
'argmin_predict.tst')
idx = np.argmax(col_dict.pop('i'))
for key, col in col_dict.iteritems():
prediction_dict[key] = col[idx]
return prediction_dict