Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 6 additions & 4 deletions atarashi/evaluator/evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,10 @@
__author__ = "Ayush Bhardwaj"
__email__ = "classicayush@gmail.com"

from pkg_resources import resource_filename
try:
from importlib.resources import files
except ImportError:
from importlib_resources import files
import argparse
import zipfile
import shutil
Expand Down Expand Up @@ -113,9 +116,8 @@ def evaluate(scanner):


if __name__ == "__main__":
defaultProcessed = resource_filename("atarashi",
"data/licenses/processedLicenses.csv")
defaultJSON = resource_filename("atarashi", "data/Ngram_keywords.json")
defaultProcessed = str(files("atarashi").joinpath("data/licenses/processedLicenses.csv"))
defaultJSON = str(files("atarashi").joinpath("data/Ngram_keywords.json"))
parser = argparse.ArgumentParser()
parser.add_argument("-a", "--agent_name", required=True,
choices=['wordFrequencySimilarity', 'DLD', 'tfidf', 'Ngram'],
Expand Down