From b13f0a3f927e52b5ccb9ec6f36064f19a1f5af9e Mon Sep 17 00:00:00 2001 From: saanikat Date: Fri, 13 Sep 2024 16:19:50 -0400 Subject: [PATCH] minor changes --- attribute_standardizer/attr_standardizer.py | 1 - attribute_standardizer/utils.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/attribute_standardizer/attr_standardizer.py b/attribute_standardizer/attr_standardizer.py index 13bf949..6825392 100644 --- a/attribute_standardizer/attr_standardizer.py +++ b/attribute_standardizer/attr_standardizer.py @@ -86,7 +86,6 @@ def _get_parameters(self) -> Tuple[int, int, int, int, int, float]: def _load_model(self) -> nn.Module: """ Calls function to load the model from HuggingFace repository and sets to eval(). - :return nn.Module: Loaded Neural Network Model. """ try: diff --git a/attribute_standardizer/utils.py b/attribute_standardizer/utils.py index aff492e..7e24487 100644 --- a/attribute_standardizer/utils.py +++ b/attribute_standardizer/utils.py @@ -186,7 +186,7 @@ def data_encoding( X_values_embeddings = embeddings if schema == "ENCODE": # Bag of Words Vectorizer - vectorizer = CountVectorizer() + vectorizer = None vc_path = hf_hub_download( repo_id=REPO_ID, filename=ENCODE_VECTORIZER_FILENAME, @@ -202,7 +202,7 @@ def data_encoding( # print(transformed_columns) X_values_bow = transformed_columns # Label Encoding - label_encoder = LabelEncoder() + label_encoder = None lb_path = hf_hub_download( repo_id=REPO_ID, filename=ENCODE_LABEL_ENCODER_FILENAME,