UMCU
/

ICD10_classifier_base_English

Feature Extraction

text-classification

text-embeddings-inference

Model card Files Files and versions

UMCU commited on Aug 30

Commit

3f87fc8

·

verified ·

1 Parent(s): 2198535

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -90,9 +90,9 @@ def get_feature_vector(text):
 def reform(features, how='softmax'):
     if how == 'softmax':
         return torch.nn.functional.softmax(features)
-    elif how = 'gumbel':
         return torch.nn.functional.gumbel_softmax(features)
-    elif how = 'normalize':
         return torch.nn.functional.normalize(features.reshape(1,-1))
 feature_vector = reform(get_feature_vector(text))

 def reform(features, how='softmax'):
     if how == 'softmax':
         return torch.nn.functional.softmax(features)
+    elif how == 'gumbel':
         return torch.nn.functional.gumbel_softmax(features)
+    elif how == 'normalize':
         return torch.nn.functional.normalize(features.reshape(1,-1))
 feature_vector = reform(get_feature_vector(text))