diff --git a/fastNLP/models/base_model.py b/fastNLP/models/base_model.py index 61edb91f..f1896cb2 100644 --- a/fastNLP/models/base_model.py +++ b/fastNLP/models/base_model.py @@ -22,6 +22,9 @@ class BaseModel(torch.nn.Module): class NaiveClassifier(BaseModel): + """ + 一个简单的分类器例子,可用于各种测试 + """ def __init__(self, in_feature_dim, out_feature_dim): super(NaiveClassifier, self).__init__() self.mlp = MLP([in_feature_dim, in_feature_dim, out_feature_dim])