Skip to content

Commit 036c146

Browse files
committed
updates
1 parent 87af712 commit 036c146

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/LLM/图解BERT.md

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,4 +196,12 @@ features.append(
196196
input_len=input_len))
197197
```
198198
> label 是当前文本对应的类别标签
199-
> input_len 是序列实际长度(含special tokens)
199+
> input_len 是序列实际长度(含special tokens)
200+
201+
11. 数据集预处理完后,将InputFeatures List列表组装起来得到需要的DataSet
202+
203+
```python
204+
dataset = TensorDataset(all_input_ids, all_attention_mask, all_token_type_ids, all_lens,all_labels)
205+
```
206+
207+
## 模型架构

0 commit comments

Comments
 (0)