update init files

fastNLP/__init__.py

@@ -10,7 +10,7 @@ fastNLP 由 :mod:`~fastNLP.core` 、 :mod:`~fastNLP.io` 、:mod:`~fastNLP.module
 
 fastNLP 中最常用的组件可以直接从 fastNLP 包中 import ，他们的文档如下：
 """
-__all__ = ["Instance", "FieldArray", "Batch", "Vocabulary", "DataSet",
+__all__ = ["Instance", "FieldArray", "Batch", "Vocabulary", "DataSet", "Const",
            "Trainer", "Tester", "Callback",
            "Padder", "AutoPadder", "EngChar2DPadder",
            "AccuracyMetric", "BMESF1PreRecMetric", "SpanFPreRecMetric", "SQuADMetric",

fastNLP/core/__init__.py

@@ -13,6 +13,8 @@ core 模块里实现了 fastNLP 的核心框架，常用的组件都可以从 fa
 
 """
 from .batch import Batch
+from .callback import Callback
+from .const import Const
 from .dataset import DataSet
 from .field import FieldArray, Padder, AutoPadder, EngChar2DPadder
 from .instance import Instance
@@ -22,6 +24,5 @@ from .optimizer import Optimizer, SGD, Adam
 from .sampler import SequentialSampler, BucketSampler, RandomSampler, Sampler
 from .tester import Tester
 from .trainer import Trainer
+from .utils import cache_results
 from .vocabulary import Vocabulary
-from .callback import Callback
-from .utils import cache_results

fastNLP/models/__init__.py

@@ -4,7 +4,10 @@ TODO 详细介绍的表格，与主页相对应
 
 """
 from .base_model import BaseModel
+from .bert import BertForMultipleChoice, BertForQuestionAnswering, BertForSequenceClassification, \
+    BertForTokenClassification
 from .biaffine_parser import BiaffineParser, GraphParser
 from .cnn_text_classification import CNNText
 from .sequence_modeling import SeqLabeling, AdvSeqLabel
 from .snli import ESIM
+from .star_transformer import STSeqCls, STNLICls, STSeqLabel

fastNLP/modules/__init__.py

@@ -16,4 +16,3 @@ from .dropout import TimestepDropout
 from .encoder import *
 from .utils import get_embeddings
 
-__version__ = '0.0.0'

fastNLP/modules/aggregator/__init__.py

@@ -1,7 +1,7 @@
-__all__ = ["MaxPool", "MaxPoolWithMask", "AvgPool", "MultiHeadAttention"]
+__all__ = ["MaxPool", "MaxPoolWithMask", "AvgPool", "MultiHeadAttention", "BiAttention"]
 from .pooling import MaxPool
 from .pooling import MaxPoolWithMask
 from .pooling import AvgPool
 from .pooling import MeanPoolWithMask
 
-from .attention import MultiHeadAttention
+from .attention import MultiHeadAttention, BiAttention

fastNLP/modules/decoder/__init__.py

@@ -1,4 +1,4 @@
-__all__ = ["MLP", "ConditionalRandomField","viterbi_decode"]
+__all__ = ["MLP", "ConditionalRandomField", "viterbi_decode"]
 from .CRF import ConditionalRandomField
 from .MLP import MLP
-from .utils import viterbi_decode
+from .utils import viterbi_decode