vt-le/astnet

training code

Opened this issue · 1 comments

Devansh-Sehrawat commented

Hi tried building a training code for this, but got only 76% AUROC.
Could you help me with the debug where it's going wrong.
repo: https://github.com/Devansh-Sehrawat/astnet/blob/main/trainMod.py
mail: davebeeta@gmail.com

vt-le commented

Hi tried building a training code for this, but got only 76% AUROC. Could you help me with the debug where it's going wrong. repo: https://github.com/Devansh-Sehrawat/astnet/blob/main/trainMod.py mail: davebeeta@gmail.com

Hi,
Thanks for your interest!
The training script of ASTNet is released.