GitHub - Dusker233/maimai-keywords-predictor

XgBoost 基准分类：acc=0.0360

BiLSTM: acc=0.812 Transformer: acc=0.8151 LSTM: acc=0.8022

highest error sample when training

Song_id	Level_index	Loss	Prediction	Label	Method
11663	4	26.32314682006836	[ 1, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0 ]	[ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 ]	BiLSTM
11663	4	31.803020477294922	[ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ]	[ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 ]	Transformer
11663	4	26.626995086669922	[ 1, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0 ]	[ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 ]	LSTM

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
Maichart-Converts		Maichart-Converts
info		info
models		models
new_data		new_data
new_maidata/world's end blackbox		new_maidata/world's end blackbox
openmdp		openmdp
preprocess		preprocess
serialized_data		serialized_data
trained_models		trained_models
utils		utils
.gitignore		.gitignore
.python-version		.python-version
README.md		README.md
__init__.py		__init__.py
combined-tags-mapped.json		combined-tags-mapped.json
combined-tags.json		combined-tags.json
consts.py		consts.py
error.log		error.log
eval_model.py		eval_model.py
eval_model_transformer.py		eval_model_transformer.py
label_getter.py		label_getter.py
pyproject.toml		pyproject.toml
serializer.py		serializer.py
tag_id_mapping.py		tag_id_mapping.py
test_xgboost.py		test_xgboost.py
title_id.csv		title_id.csv
title_id_getter.py		title_id_getter.py
train.py		train.py
train_bilstm.py		train_bilstm.py
train_transformer.py		train_transformer.py
uv.lock		uv.lock