Skip to content

Adding new tf_tools and new conditional deep sets model

Alexander Froch requested to merge alfroch-adding-DIPS-attention into preprocessing-remake

This MR restructures the definition of the neural network models. All tensorflow/keras related functions (model definitions, custom layers, generators) are now in a separate folder `umami/tf_tools.

Also the new conditional deep sets model + all needed custom layers from @mguth and @jraine will be added in this step.

Edited by Alexander Froch

Merge request reports