Skip to content

Latest commit

 

History

History
89 lines (59 loc) · 1.75 KB

README.md

File metadata and controls

89 lines (59 loc) · 1.75 KB
Raw

IND-WhoIsWho

Team DeepMayNotLearn Rank 11

Prerequisites

  • Ubuntu22.04
  • Python 3.10
  • PyTorch 2.2.0+cu121

Hardware device

  • GPU 3090 24G
  • RAM 64G

Run Code

Run GNN

install the requirements

sh GNN/install_requirements.sh

build the graph ,and this may cost lots of time,if need you can find me to get my built graph

sh GNN/build_graph.sh

train and save the model

sh GNN/train_model.sh

predict

python GNN/w2v_model_test.py

python GNN/sci_model_test.py

To perfectly reproduce the effect, you can load the trained model for inference using the following code

python GNN/trained_w2v_model_test.py

python GNN/trained_sci_model_test.py

Run LGB+XGB+CAT

sh lgb+cat+xgb/install_requires.sh

train_w2v

python lgb+cat+xgb/w2vProcess.py

train-and-predict

python lgb+cat+xgb/main.py

Run Melt To Get The Final Result

python melt.py

Resourece

IND Dataset:Data (biendata.xyz)

SciBERT model:allenai/scibert_scivocab_uncased at main (hf-mirror.com)

Trained Model or More externel data:链接：https://pan.baidu.com/s/1toWSZvdgm-ZTEBogzgNJsw 提取码：deep

Methods

You can see more details in GNN/README.md and lgb+cat+xgb/README.md

Team Introduction

  • DeepMayNotLearn, junior student of CQUPT
  • yueming, junior student of CQUPT
  • yxw666, junior student of CQUPT