search

realHXiao / GraphKM

KM prediction model
0 stars 1 forks source link

Help：no trained model gin_xgboost_model.dat #1

Open AaroncrowAries opened 1 week ago

AaroncrowAries commented 1 week ago

Dear Xiao He,

I have noticed during my usage that the pre-trained model file best_modelgin-1_lr0.0005.pdparams and gin_xgboost_model.dat are not provided. Since my PC is not powerful enough to retrain the model, I was wondering if it is possible for you to release the already trained model?

Additionally, there seems to be an error in the example command. It could be modified to: "python prediction.py -c --csv_file example.csv -l KM --input_seq example.tsv -m path_to/best_modelgin-1_lr0.0005.pdparams --model_config ./Train_codes/model_configs/gin_config.json -xgb path_to/gin_xgboost_model.dat"

Thank you for your attention to this matter.

Best regards,

realHXiao commented 1 week ago

gin_xgboost_model.dat is released in the Figshare database with https://doi.org/10.6084/m9.figshare.25335049.