-
Notifications
You must be signed in to change notification settings - Fork 94
Open
Description
in dataset/base.py file, this line:
smap = {s: i for i, s in enumerate(set(df['sid']))}
should be changed to:
smap = {s: i + 1 for i, s in enumerate(set(df['sid']))}
because enumerate starts from 0 and bert model ignores 0. 0 indexed anime will be considered from the padded part of the vector and will be ignored. to avoid that, we should start from index 1.
Metadata
Metadata
Assignees
Labels
No labels