Imported from https://github.com/mattsherar/Temporal_Fusion_Transform