LICENSE.md
README.md
pyproject.toml
ritini/__init__.py
ritini/train.py
ritini.egg-info/PKG-INFO
ritini.egg-info/SOURCES.txt
ritini.egg-info/dependency_links.txt
ritini.egg-info/requires.txt
ritini.egg-info/top_level.txt
ritini/data/__init__.py
ritini/data/temporal_graph.py
ritini/data/trajectory_loader.py
ritini/data_generation/__init__.py
ritini/data_generation/toy_data.py
ritini/models/RiTINI.py
ritini/models/__init__.py
ritini/models/gat.py
ritini/models/gatConvwithAttention.py
ritini/models/graphDifferentialEquation.py
ritini/models/meanAttentionLayer.py
ritini/models/ode.py
ritini/models/time_attention.py
ritini/utils/__init__.py
ritini/utils/attention_graphs.py
ritini/utils/preprocess.py
ritini/utils/prior_graph.py
ritini/utils/utils.py
ritini/utils/visualization.py
ritini/visualizations/__init__.py
ritini/visualizations/graph_visualizations.py
ritini/visualizations/visualize_dynamics_prediction.py
ritini/visualizations/visualize_graphs_attention_gat.py
ritini/visualizations/visualize_graphs_attention_ritini.py
ritini/visualizations/visualize_graphs_attention_ritini_toy.py
ritini/visualizations/visualize_graphs_temporal_gat.py
ritini/visualizations/visualize_graphs_toy_data.py
ritini/visualizations/visualize_natalia_ritini.py
ritini/visualizations/visualize_toy_data.py
tests/test_real_data_gat.py
tests/test_toy_data_gat.py
tests/test_toy_data_ritini.py