|
|
|
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) |
|
|