LT-OCF: Learnable-Time ODE-based Collaborative Filtering
Our proposed LT-OCF
Our proposed dual co-evolving ODE
Setup Python environment for LT-OCF
Install python environment
conda env create -f environment.yml
Activate environment
Reproducibility
Usage
In terminal
Run the shell file (at the root of the project)
# run lt-ocf (gowalla dataset, rk4 solver, learnable time)
sh ltocf_gowalla_rk4.sh
# run lt-ocf (gowalla dataset, rk4 solver, fixed time)
sh ltocf_gowalla_rk4_fixed.sh
Arguments (see more arguments in parse.py
)
gpuid
dataset
gowalla, yelp2018, amazon-book
model
solver
euler, rk4 , implicit_adams, dopri5
adjoint
K
learnable_time
dual_res