search

webis-de / small-text

Active Learning for Text Classification in Python
https://small-text.readthedocs.io/
MIT License
547 stars 60 forks source link

Inconsistency: TransformerBaseClassification.embed() returns np.float32 while SetFitClassification returns np.float64 #48

Open chschroeder opened 11 months ago

chschroeder commented 11 months ago

Desired Solution

Clearly specify the EmbeddingMixin and make all implementations behave consistently. All returned embeddings should be np.float32.

This fits in well with the ongoing typing efforts.