| | |
| | import matplotlib.pyplot as plt |
| | import os |
| | import csv |
| |
|
| | labels = [] |
| | numbers = [] |
| | numEntries = 1 |
| |
|
| | rows = [] |
| |
|
| |
|
| | def bar_chart(numbers, labels, pos): |
| | plt.bar(pos, numbers, color='blue') |
| | plt.xticks(ticks=pos, labels=labels) |
| | plt.title("Jeopardy Results by Model") |
| | plt.xlabel("Model") |
| | plt.ylabel("Questions Correct") |
| | plt.show() |
| |
|
| |
|
| | def calculatecorrect(): |
| | directory = os.fsencode("./examples/jeopardy/results/") |
| | csv_reader = csv.reader(open("./examples/jeopardy/qasheet.csv", 'rt'), delimiter=',') |
| | for row in csv_reader: |
| | global rows |
| | rows.append(row) |
| | for listing in os.listdir(directory): |
| | filename = os.fsdecode(listing) |
| | if filename.endswith(".txt"): |
| | file = open("./examples/jeopardy/results/" + filename, "rt") |
| | global labels |
| | global numEntries |
| | global numbers |
| | labels.append(filename[:-4]) |
| | numEntries += 1 |
| | i = 1 |
| | totalcorrect = 0 |
| | for line in file.readlines(): |
| | if line.strip() != "------": |
| | print(line) |
| | else: |
| | print("Correct answer: " + rows[i][2] + "\n") |
| | i += 1 |
| | print("Did the AI get the question right? (y/n)") |
| | if input() == "y": |
| | totalcorrect += 1 |
| | numbers.append(totalcorrect) |
| |
|
| |
|
| | if __name__ == '__main__': |
| | calculatecorrect() |
| | pos = list(range(numEntries)) |
| | labels.append("Human") |
| | numbers.append(48.11) |
| | bar_chart(numbers, labels, pos) |
| | print(labels) |
| | print(numbers) |
| |
|
