Release v0.4.5

luozhouyang · Sep 7, 2021 · 5a63412 · 5a63412
1 parent a950139
commit 5a63412
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="transformers_keras",
-    version="0.4.4",
+    version="0.4.5",
     description="Transformer-based models implemented in tensorflow 2.x(Keras)",
     long_description=long_description,
     long_description_content_type="text/markdown",

diff --git a/transformers_keras/__init__.py b/transformers_keras/__init__.py
@@ -40,7 +40,22 @@
 )
 from transformers_keras.sentence_embedding.bert_embedding import BertForSentenceEmbedding
 from transformers_keras.sentence_embedding.callback import SpearmanForSentenceEmbedding
+from transformers_keras.sentence_embedding.simcse_dataset import (
+    HardNegativeSimCSEDataset,
+    HardNegativeSimCSEExample,
+    SupervisedSimCSEDataset,
+    SupervisedSimCSEExample,
+    UnsupervisedSimCSEDataset,
+    UnsupervisedSimCSEExample,
+)
 from transformers_keras.sentence_embedding.simcse_models import HardNegativeSimCSE, SupervisedSimCSE, UnsupervisedSimCSE
+from transformers_keras.sentiment_analysis.ate import BertForAspectTermExtraction
+from transformers_keras.sentiment_analysis.dataset import (
+    AspectTermExtractionDataset,
+    AspectTermExtractionExample,
+    OpinionTermExtractionAndClassificationDataset,
+)
+from transformers_keras.sentiment_analysis.otec import BertForOpinionTermExtractionAndClassification
 from transformers_keras.sequence_classification.dataset import SequenceClassificationDataset, SequenceClassificationExample
 from transformers_keras.sequence_classification.models import AlbertForSequenceClassification, BertForSequenceClassification
 from transformers_keras.token_classification.callback import SeqEvalForTokenClassification
@@ -53,6 +68,6 @@
 from transformers_keras.token_classification.models import AlbertForTokenClassification, BertForTokenClassification
 
 __name__ = "transformers_keras"
-__version__ = "0.4.4"
+__version__ = "0.4.5"
 
 logging.basicConfig(format="%(asctime)s %(levelname)7s %(filename)20s %(lineno)4d] %(message)s", level=logging.INFO)