Skip to content

Commit eb5025a

Browse files
committed
Release v0.4.2
1 parent f8cc46d commit eb5025a

File tree

2 files changed

+30
-3
lines changed

2 files changed

+30
-3
lines changed

setup.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
setuptools.setup(
77
name="transformers_keras",
8-
version="0.4.1",
8+
version="0.4.2",
99
description="Transformer-based models implemented in tensorflow 2.x(Keras)",
1010
long_description=long_description,
1111
long_description_content_type="text/markdown",
@@ -15,7 +15,8 @@
1515
packages=setuptools.find_packages(),
1616
include_package_data=True,
1717
install_requires=[
18-
18+
"tokenizers",
19+
"tensorflow>=2.0.0",
1920
],
2021
extras_require={
2122

transformers_keras/__init__.py

Lines changed: 27 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,20 +21,46 @@
2121
BertPretrainedModel,
2222
)
2323
from transformers_keras.modeling_utils import complete_inputs
24+
from transformers_keras.question_answering.dataset import QuestionAnsweringDataset, QuestionAnsweringExample
2425
from transformers_keras.question_answering.models import AlbertForQuestionAnswering, BertForQuestionAnswering
26+
from transformers_keras.question_answering.parser import (
27+
AbstractQustionAnsweringExampleParser,
28+
QuestionAnsweringExampleParserForChinese,
29+
)
30+
from transformers_keras.question_answering.tokenizer import (
31+
QuestionAnsweringTokenizer,
32+
QuestionAnsweringTokenizerForChinese,
33+
)
34+
from transformers_keras.sentence_embedding.bert_embedding import BertForSentenceEmbedding
35+
from transformers_keras.sentence_embedding.parser import AbstractSimCSEExampleParser, SimCSEExampleParser
36+
from transformers_keras.sentence_embedding.simcse_dataset import SimCSEDataset, SimCSEExample
2537
from transformers_keras.sentence_embedding.simcse_models import HardNegativeSimCSE, SupervisedSimCSE, UnsupervisedSimCSE
38+
from transformers_keras.sequence_classification.dataset import (
39+
SequenceClassificationDataset,
40+
SequenceClassificationExample,
41+
)
2642
from transformers_keras.sequence_classification.models import (
2743
AlbertForSequenceClassification,
2844
BertForSequenceClassification,
2945
)
46+
from transformers_keras.sequence_classification.parser import (
47+
AbstractSequenceClassificationExampleParser,
48+
SequenceClassificationExampleParser,
49+
)
3050
from transformers_keras.token_classification.crf_models import (
3151
AlertCRFForTokenClassification,
3252
BertCRFForTokenClassification,
3353
CRFModel,
3454
)
55+
from transformers_keras.token_classification.dataset import TokenClassificationDataset, TokenClassificationExample
3556
from transformers_keras.token_classification.models import AlbertForTokenClassification, BertForTokenClassification
57+
from transformers_keras.token_classification.parser import (
58+
AbstractTokenClassificationExampleParser,
59+
TokenClassificationExampleParserForChinese,
60+
)
61+
from transformers_keras.token_classification.tokenizer import QuestionAnsweringTokenizerForChinese
3662

3763
__name__ = "transformers_keras"
38-
__version__ = "0.4.1"
64+
__version__ = "0.4.2"
3965

4066
logging.basicConfig(format="%(asctime)s %(levelname)7s %(filename)20s %(lineno)4d] %(message)s", level=logging.INFO)

0 commit comments

Comments
 (0)