Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
hellonlp authored Jul 12, 2023
1 parent d58da5a commit 569a523
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@
```

### 1. classifier_multi_label
<img src="https://github.com/hellonlp/classifier-multi-label/blob/master/imgs/base.png" width="700" height="200">

- 使用BERT第一个token[CLS]的向量,维度为(batch_size,hidden_size)。
- 使用了tf.nn.sigmoid_cross_entropy_with_logits的损失函数。
- 使用了tf.where函数来选择概率小于0.5的对应id。
Expand Down

0 comments on commit 569a523

Please sign in to comment.