Support loading pre-trained model parts

tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning

Apache License 2.0

3.18k stars 439 forks source link

Support loading pre-trained model parts #122

Closed HiuNaoki closed 8 years ago

HiuNaoki commented 8 years ago

i want to using autoencoder for pre-training in skflow....

please help me..

ilblackdragon commented 8 years ago

Currently it's not very simple at all. The easiest way, would be to dump numpy arrays from the pre-trained model (using get_tensor_value method on estimator) and then use initializer for your variables in your target model.

I'll make it easier, but sklearn doesn't have anything remote to API for this, so will need to think carefully to have something that would be useable in years to come.

ilblackdragon commented 8 years ago

Here is suggested API: https://github.com/tensorflow/tensorflow/issues/2141 and will track this functionality there.