Tensorflow 2.0 Snippets for VS Code

Type prefix tfk: in Python file to get start with this extension! ( tfk presents tensorflow keras)

This extension includes some snippets to develop a model in your VS Code editor. It requires the minimum version of tensorflow is 2.0. The inside keras package will be used in this extension. For more information, please go to Key Features .

Support

While being free and open source, if you find it useful, please consider star us on the repository tensorflow2snippets .

Getting started

The extension requires some usefull packages, such as tensorflow tensorflow-datasets tf-nightly.

pip install tensorflow tf-nightly 
pip install tensorflow-datasets

The visualization tool and some auxiliary tools are also important for you to train a deep learning model.

pip install matplotlib numpy

Key Features

tfk:dataset Field

We write some snippets to help you load the data set easily. But first you must import tensorflow_datasets.

import tensorflow_datasets as tfds

# Image Classification data set
tfk:dataset:mnist 
tfk:dataset:fmnist
tfk:dataset:cifar10 
tfk:dataset:cifar100 
tfk:dataset:imagenet2012
tfk:dataset:celeb_a 

# Object_detection data set
tfk:dataset:coco
tfk:dataset:voc2007

# Structured data set
tfk:dataset:iris

# Generative data set
tfk:dataset:cycle_gan

# Summarization data set
tfk:dataset:scientific_papers

# Text Classification data set
tfk:dataset:scicite

# Language Translate data set
tfk:dataset:flores

# Movies data set
tfk:dataset:moving_mnist

tfk:ctrl Field

This field can help you control the procedure flow more convenience. They generate function like model.fit(...) or class like class model(tf.keras.Model).

tfk:ctrl:model
tfk:ctrl:fit
tfk:ctrl:evaluate
tfk:ctrl:compile
tfk:ctrl:callback
tfk:ctrl:saveModel
tfk:ctrl:loadModel

tfk:code Field

This field can generate a runnable .py file. It can used to set the code frame. All of the codes are derived from tensorflow official guide. We express our appreciation for tensorflow developers and the author of the official document.

tfk:code:mnist:simple # Simple code for img classification
tfk:code:mnist:full # More complex code for img classification
tfk:code:oxford # Code for img segmentation
tfk:code:translate # Code for NLP-translate
tfk:code:word_embeddings # Code for NLP-word embeddings

How to Contribute

This project welcomes contributions and suggestions.

Fork our repository and you can submit the snippet you like.

Bug Report or Problem

If you find some bugs or have some problems on this extension. Please create your issue.

Contributors

Kaiyan Chang

License