SafeSpaceAI / src /evaluation /evaluate.py
Shreya Pal
Make API Key private
5c5b473
raw
history blame contribute delete
238 Bytes
def evaluate(agent, env):
correct = 0
total = len(env.data)
for text, label in env.data:
action = agent.choose_action(text)
if action == label:
correct += 1
print("Accuracy:", correct / total)