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 loading and Keras Model from `.h5` file #68

Open basedrhys opened 2 years ago

basedrhys commented 2 years ago

Is your feature request related to a problem? Please describe. WDL4J contains a number of pre-loaded neural networks that can be used out of the box, however, I may have trained my own network using Keras and it would be nice to be able to load the .h5 file into WDL4J

Describe the solution you'd like A file selector box and functionality to load a keras model from .h5 file