taesiri's picture
initial commit
a8ed63e
raw history blame
No virus
2.98 kB
from glob import glob
import json
import numpy as np
import gradio as gr
def calculate_the_results():
all_jsons_path = glob('./responses/*.json')
all_jsons = [json.load(open(path)) for path in all_jsons_path]
# count number of user corrects for each json and average and also calcaulte the type of NNs
top1_results = []
top1_acc = []
topK_results = []
topK_acc = []
for js in all_jsons:
# read one key and determine the type of NN
type_of_NNs = js['history'][0]['type']
if type_of_NNs == 'topK':
acc = np.mean([js['history'][x]['is_user_correct'] for x in range(len(js['history']))])
topK_acc.append((acc*100).round(2))
topK_results.append(js)
else:
top1_results.append(js)
acc = np.mean([js['history'][x]['is_user_correct'] for x in range(len(js['history']))])
top1_acc.append((acc*100).round(2))
print('# of top1: ', len(top1_results))
print('top1 Accuracy: ', top1_acc)
# print std and mean of top1_acc
std = np.std(top1_acc)
mean = np.mean(top1_acc)
print('top1 std: ', std)
print('top1 mean: ', mean)
print('----------------------------------')
print('# of topK: ', len(topK_results))
print('topK Accuracy: ', topK_acc)
std = np.std(topK_acc)
mean = np.mean(topK_acc)
print('topK std: ', std)
print('topK mean: ', mean)
def calculate_the_results():
all_jsons_path = glob('./responses/*.json')
all_jsons = [json.load(open(path)) for path in all_jsons_path]
# count number of user corrects for each json and average and also calculate the type of NNs
top1_results = []
top1_acc = []
topK_results = []
topK_acc = []
for js in all_jsons:
# read one key and determine the type of NN
type_of_NNs = js['history'][0]['type']
if type_of_NNs == 'topK':
acc = np.mean([js['history'][x]['is_user_correct'] for x in range(len(js['history']))])
topK_acc.append((acc*100).round(2))
topK_results.append(js)
else:
top1_results.append(js)
acc = np.mean([js['history'][x]['is_user_correct'] for x in range(len(js['history']))])
top1_acc.append((acc*100).round(2))
top1_output = f"# of top1: {len(top1_results)}\ntop1 Accuracy: {top1_acc}\ntop1 std: {np.std(top1_acc)}\ntop1 mean: {np.mean(top1_acc)}\n----------------------------------\n"
topK_output = f"# of topK: {len(topK_results)}\ntopK Accuracy: {topK_acc}\ntopK std: {np.std(topK_acc)}\ntopK mean: {np.mean(topK_acc)}"
return top1_output + topK_output
with gr.Blocks(theme=gr.themes.Soft()) as demo:
update_btn = gr.Button("Calculate the results")
results_textbox = gr.Textbox(lines=10, label="Results")
update_btn.click(fn=calculate_the_results, outputs=results_textbox)
demo.launch(debug=False, server_name="0.0.0.0", server_port=9911)