27 lines
621 B
Python
27 lines
621 B
Python
import os
|
|
import re
|
|
import pandas as pd
|
|
import numpy as np
|
|
from chatbot.modules.nlu import NLU
|
|
|
|
rows = 0
|
|
hits = 0
|
|
|
|
nlu = NLU()
|
|
|
|
for file_name in os.listdir("data"):
|
|
df = pd.read_csv(f"data/{file_name}", sep="\t", names=["user", "sentence", "acts"])
|
|
df = df[df.user == "user"]
|
|
data = np.array(df)
|
|
|
|
for row in data:
|
|
rows += 1
|
|
sentence = row[1]
|
|
cleaned_text = re.sub(r'\([^)]*\)', '', row[2])
|
|
user_acts = cleaned_text.split("&")
|
|
nlu_match = nlu.match(sentence)
|
|
if nlu_match["act"] in user_acts:
|
|
hits += 1
|
|
|
|
print(f"Accuracy: {(hits / rows) * 100}")
|