Refactor target models
bentsherman opened this issue · comments
The three target model classes can be reduced to a single class with arguments to control hiddlen layer sizes and batch norm. Additionally, this code should be updated to use Keras API as the target model is a standard MLP.
Done in recent commits.