This repository contains the code base for "The Beran Estimator with Neural Kernels for Estimating the Heterogeneous Treatment Effect" article. The program is written in python 3 using the Pytorch framework.
To satisfy the dependencies use pip and requirements.txt:
pip install -r requirements.txtView example for the brief info.
The pytorch_survival script contains all numerical experiments, described in the article. To visualize collected data use the results merging script to merge all the experiments results and then draw the figures using the drawing script.