Browse Source

add doc for NaiveClassifier

tags/v0.4.10
ChenXin 5 years ago
parent
commit
8f0f280629
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      fastNLP/models/base_model.py

+ 3
- 0
fastNLP/models/base_model.py View File

@@ -22,6 +22,9 @@ class BaseModel(torch.nn.Module):




class NaiveClassifier(BaseModel): class NaiveClassifier(BaseModel):
"""
一个简单的分类器例子,可用于各种测试
"""
def __init__(self, in_feature_dim, out_feature_dim): def __init__(self, in_feature_dim, out_feature_dim):
super(NaiveClassifier, self).__init__() super(NaiveClassifier, self).__init__()
self.mlp = MLP([in_feature_dim, in_feature_dim, out_feature_dim]) self.mlp = MLP([in_feature_dim, in_feature_dim, out_feature_dim])


Loading…
Cancel
Save