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