Fix to return proper values

create.py

@@ -35,7 +35,8 @@ def create(text,score,prompt_string):
 
     #Initialize a results dictionary to return
     results = {'errors': [],'success' : False, 'cv_kappa' : 0, 'cv_mean_absolute_error': 0,
-               'feature_ext' : "", 'classifier' : "", 'algorithm' : util_functions.AlgorithmTypes.classification}
+               'feature_ext' : "", 'classifier' : "", 'algorithm' : util_functions.AlgorithmTypes.classification,
+               'score' : score, 'text' : text, 'prompt' : prompt_string}
 
     if len(text)!=len(score):
         msg = "Target and text lists must be same length."

grade.py

@@ -69,7 +69,7 @@ def grade(grader_data,submission):
 
     #Try to determine confidence level
     try:
-        results['confidence'] = get_confidence_value(grader_data['algorithm'], grader_data['model'], grader_feats, results['score'])
+        results['confidence'] = get_confidence_value(grader_data['algorithm'], grader_data['model'], grader_feats, results['score'], grader_data['score'])
     except:
         #If there is an error getting confidence, it is not a show-stopper, so just log
         log.exception("Problem generating confidence value")