Description
Interface to 'TensorFlow' Hub.
Description
'TensorFlow' Hub is a library for the publication, discovery, and consumption of reusable parts of machine learning models. A module is a self-contained piece of a 'TensorFlow' graph, along with its weights and assets, that can be reused across different tasks in a process known as transfer learning. Transfer learning train a model with a smaller dataset, improve generalization, and speed up training.
README.md
tfhub
The tfhub package provides R wrappers to TensorFlow Hub.
TensorFlow Hub is a library for reusable machine learning modules.
TensorFlow Hub is a library for the publication, discovery, and consumption of reusable parts of machine learning models. A module is a self-contained piece of a TensorFlow graph, along with its weights and assets, that can be reused across different tasks in a process known as transfer learning. Transfer learning can:
- Train a model with a smaller dataset,
- Improve generalization, and
- Speed up training.
Installation
You can install the development version from GitHub with:
# install.packages("devtools")
devtools::install_github("rstudio/tfhub")
After installing the tfhub package you need to install the TensorFlow Hub python module:
library(tfhub)
install_tfhub()
Go to the website for more information.