ydeng/mltraining
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Restructured in preparation for packaging

Browse Source
This commit is contained in:
Harrison Deng 2023-06-01 16:16:13 -05:00
parent d6c0130477
commit 80a62a6e32
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trainers.py → src/mltraining/trainers.py
View File

@ -57,7 +57,7 @@ class TokenClassificationTrainer:
) )
labels = [] labels = []
for i, label in enumerate(examples[f"ner_tags"]): for i, label in enumerate(examples["ner_tags"]):
word_ids = tokenized_inputs.word_ids( word_ids = tokenized_inputs.word_ids(
batch_index=i batch_index=i
) # Map tokens to their respective word. ) # Map tokens to their respective word.