Skip to content

a thin scikit-learn style wrapper for tensorflow framework

Notifications You must be signed in to change notification settings

DSLituiev/tf_sklearn_wrapper

Repository files navigation

tflearn

a thin scikit-learn style wrapper for tensorflow framework

Usage

The wrapper takes care of training (fit()) and prediction(predict()) infrastructure. What is left to do is to specify the network topology in _create_network() method and loss function in _create_loss() method.

There are two subclasses of the tflearn class which come with ready to use _create_loss() method:

  • rtflearn for regression
  • ctflearn for classification (in progress)

Examples

tf_lasso.py: Lasso regression, a simplest example

tf_factorization_machine.ipynb : Factorization machines

gan_dense.py : Generative adversarial network with dense perceptron layers

About

a thin scikit-learn style wrapper for tensorflow framework

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published