From 50b34095782de47d8098efc77a6d1a3a3275dcb8 Mon Sep 17 00:00:00 2001 From: Pavan Mandava Date: Sat, 1 Aug 2020 14:04:27 +0200 Subject: [PATCH] intent_predictor moved to classifier package --- README.md | 2 +- classifier/__init__.py | 1 + {testing => classifier}/intent_predictor.py | 0 testing/bilstm_predict.py | 4 +--- 4 files changed, 3 insertions(+), 4 deletions(-) rename {testing => classifier}/intent_predictor.py (100%) diff --git a/README.md b/README.md index e7682a3..e921f5c 100644 --- a/README.md +++ b/README.md @@ -148,7 +148,7 @@ Our BiLSTM AllenNLP model contains 4 major components: - Dropout - Embeddings - All the classes that the Config file uses must register using Python decorators (for example, `@Model.register('bilstm_classifier'`). - 4. Predictor - [IntentClassificationPredictor](/testing/intent_predictor.py) + 4. Predictor - [IntentClassificationPredictor](/classifier/intent_predictor.py) - AllenNLP uses `Predictor`, a wrapper around the trained model, for making predictions. - The Predictor uses a pre-trained/saved model and dataset reader to predict new Instances diff --git a/classifier/__init__.py b/classifier/__init__.py index 81d9d69..4e9a149 100644 --- a/classifier/__init__.py +++ b/classifier/__init__.py @@ -1,2 +1,3 @@ from .nn import * from utils.reader import * +from.intent_predictor import * \ No newline at end of file diff --git a/testing/intent_predictor.py b/classifier/intent_predictor.py similarity index 100% rename from testing/intent_predictor.py rename to classifier/intent_predictor.py diff --git a/testing/bilstm_predict.py b/testing/bilstm_predict.py index 04dbf4f..239b19d 100644 --- a/testing/bilstm_predict.py +++ b/testing/bilstm_predict.py @@ -1,6 +1,4 @@ -import classifier - -import testing.intent_predictor as pred +import classifier.intent_predictor as pred import eval.metrics as metrics