search

julrog / nn_vis

A project for processing neural networks and rendering to gain insights on the architecture and parameters of a model through a decluttered representation.
MIT License
1.09k stars 186 forks source link

How to import a .hdf5 model file to run nn_vis ? #24

Open rkuo2000 opened 3 years ago

rkuo2000 commented 3 years ago

As title, anyway to import tensorflow.keras pretrained model file ? can it run to show visualization ?

julrog commented 3 years ago

Right now you have to code a bit for that. For calculating importance of a network, a fraction of the training/test data must be used, depending on the network, but most likely less than 1%. I will add some documentation on this in the future. Unfortunately to automate the whole process with data is difficult, because most data need specific preprocessing etc. I will look into providing some scripts to automate as much as possible. I will let you know when I made some progress.