diff --git a/interactive_predict.py b/interactive_predict.py index 039c037..f607085 100644 --- a/interactive_predict.py +++ b/interactive_predict.py @@ -2,7 +2,6 @@ from extractor import Extractor SHOW_TOP_CONTEXTS = 10 -MAX_PATH_LENGTH = 8 MAX_PATH_WIDTH = 2 EXTRACTION_API = 'https://po3g2dx2qa.execute-api.us-east-1.amazonaws.com/production/extractmethods' @@ -14,7 +13,7 @@ def __init__(self, config, model): model.predict([]) self.model = model self.config = config - self.path_extractor = Extractor(config, EXTRACTION_API, self.config.MAX_PATH_LENGTH, max_path_width=2) + self.path_extractor = Extractor(config, EXTRACTION_API, self.config.MAX_PATH_LENGTH, max_path_width=MAX_PATH_WIDTH) @staticmethod def read_file(input_filename):