Name		Name	Last commit message	Last commit date
README.md		README.md
data_utlis.py		data_utlis.py
estimators.ipynb		estimators.ipynb
eval_utlis.py		eval_utlis.py
interv_utlis.py		interv_utlis.py
main.ipynb		main.ipynb
model.py		model.py

Repository files navigation

HD2AT

Requirements

python==3.7.13
torch==1.8.0
sklearn==1.0.2
talib==0.4.19
numpy==1.21.6
pandas==1.3.5

How to train the model

Run estimators.ipynb. This script would run the preprocessing for raw data and construct the intervention graphs.
Run main.ipynb. This script would build an HD2AT model, and then train and evaluate the model.

About

No description, website, or topics provided.

3 stars

1 watching

Report repository

Releases

Packages

Contributors

Languages