Skip to content

Commit

Permalink
Merge pull request #62 from Kaushl2208/fix_evaluator
Browse files Browse the repository at this point in the history
Fix(evaluator): Fixed evaluator commands
  • Loading branch information
hastagAB authored Nov 6, 2019
2 parents 73563a4 + bf344cf commit 4064aaf
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion atarashi/evaluator/evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,8 @@ def getCommand(agent_name, similarity):
command = "atarashi -a tfidf -s CosineSim"
elif similarity == "ScoreSim":
command = "atarashi -a tfidf -s ScoreSim"
elif similarity == " ":
command = "atarashi -a tfidf"
else:
print("Please choose similarity from {CosineSim,ScoreSim}")
return -1
Expand All @@ -63,6 +65,8 @@ def getCommand(agent_name, similarity):
command = "atarashi -a Ngram -s DiceSim"
elif similarity == "BigramCosineSim":
command = "atarashi -a Ngram -s BigramCosineSim"
elif similarity == " ":
command = "atarashi -a Ngram"
else:
print("Please choose similarity from {CosineSim,ScoreSim}")
return -1
Expand Down Expand Up @@ -119,7 +123,7 @@ def evaluate(command):
parser.add_argument("-a", "--agent_name", required=True,
choices=['wordFrequencySimilarity', 'DLD', 'tfidf', 'Ngram'], help="Name of the agent that you want to evaluate")
parser.add_argument("-s", "--similarity", required=False,
default="CosineSim", choices=["ScoreSim", "CosineSim", "DiceSim", "BigramCosineSim"], help="Specify the similarity algorithm that you want to evaluate"
default=" ", choices=["ScoreSim", "CosineSim", "DiceSim", " ", "BigramCosineSim"], help="Specify the similarity algorithm that you want to evaluate"
" First 2 are for TFIDF and last 3 are for Ngram")
args = parser.parse_args()
agent_name = args.agent_name
Expand Down

0 comments on commit 4064aaf

Please sign in to comment.