From 8f0f280629fa6d95a0f2bef57cbe28869584b9c1 Mon Sep 17 00:00:00 2001 From: ChenXin Date: Mon, 23 Sep 2019 10:24:04 +0800 Subject: [PATCH] add doc for NaiveClassifier --- fastNLP/models/base_model.py | 3 +++ 1 file changed, 3 insertions(+) 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])