It's very good, I have used the repo source codes.

[yyaadet]

If you can structure code within train and predict and data load parts, or use more classes, make it easy to use and modify.

I have create a new predict.py to. handle prediction.

[bitszhang]

If you can structure code within train and predict and data load parts, or use more classes, make it easy to use and modify.

I have create a new predict.py to. handle prediction.

Can you share your code? I would like to use it, but it lacks the data construction process and prediction process

[willxxy]

If you can structure code within train and predict and data load parts, or use more classes, make it easy to use and modify. I have create a new predict.py to. handle prediction.

Can you share your code? I would like to use it, but it lacks the data construction process and prediction process

I second this. If you are willing, please share the code. Thank you!