Skip to content

Commit

Permalink
loading vectorizer and label_enc in _load_model
Browse files Browse the repository at this point in the history
  • Loading branch information
saanikat committed Sep 16, 2024
2 parents 50bc889 + b13f0a3 commit 93bca5b
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions bedms/attr_standardizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,8 +101,12 @@ def _get_parameters(self) -> Tuple[int, int, int, int, int, float]:

def _load_model(self) -> tuple[nn.Module, object, object]:
"""
<<<<<<< HEAD:bedms/attr_standardizer.py
Calls function to load the model from HuggingFace repository
load vectorizer and label encoder and sets to eval().
=======
Calls function to load the model from HuggingFace repository and sets to eval().
>>>>>>> b13f0a3f927e52b5ccb9ec6f36064f19a1f5af9e:attribute_standardizer/attr_standardizer.py
:return nn.Module: Loaded Neural Network Model.
:return object: The scikit learn vectorizer for bag of words encoding.
:return object: Label encoder object for the labels (y).
Expand Down

0 comments on commit 93bca5b

Please sign in to comment.