Can we have api for calling the model?

[HelloWorldLTY]

Hi, thanks for your great work! I wonder if we can have an API to call and train the model like what GEARS did. Thanks a lot.

[MartinZoey]

+11111111111111111111

[MartinZoey]

I modified the parameters in run.sh: --train False --no_test True --predict True. But it still occurs many bugs. It seems that the author didn't provide the realistic implement of predict mode.

[RemyLau]

Hi @HelloWorldLTY and @MartinZoey , thank you so much for your interest in our work! Really appreciate it. We will work on improving the API to make it more user friendly for training/inferencing on user specific settings besides reproducing our experiments.

Will keep you posted on the progress here, and also feel free to provide more details about how you would like the API to be.