Waikato / wekaDeeplearning4j

Weka package for the Deeplearning4j java library
https://deeplearning.cms.waikato.ac.nz/
GNU General Public License v3.0
184 stars 197 forks source link

Add support for non-linear Zoo Models #32

Open braun-steven opened 6 years ago

braun-steven commented 6 years ago

Is your feature request related to a problem? Please describe. ZooModels from Deeplearning4j such as UNet, NASNet etc. have a graph-based, non-linear architecture. Since the Dl4jMlpClassifier holds all layers in a linear/sequential manner, it is not capable of parsing these models.

Idea Create a Dl4jZooModel classifier that mimics the Dl4jMlpClassifier behavior and can load arbitrary zoo models but does not allow to change the model architecture, except for the output and input.