Convolutional Neural Network for Text Classification in Tensorflow

Last update: Jan 02, 2023

Overview

This code belongs to the "Implementing a CNN for Text Classification in Tensorflow" blog post.

It is slightly simplified implementation of Kim's Convolutional Neural Networks for Sentence Classification paper in Tensorflow.

Requirements

Python 3
Tensorflow > 0.12
Numpy

Training

Print parameters:

./train.py --help

optional arguments:
  -h, --help            show this help message and exit
  --embedding_dim EMBEDDING_DIM
                        Dimensionality of character embedding (default: 128)
  --filter_sizes FILTER_SIZES
                        Comma-separated filter sizes (default: '3,4,5')
  --num_filters NUM_FILTERS
                        Number of filters per filter size (default: 128)
  --l2_reg_lambda L2_REG_LAMBDA
                        L2 regularizaion lambda (default: 0.0)
  --dropout_keep_prob DROPOUT_KEEP_PROB
                        Dropout keep probability (default: 0.5)
  --batch_size BATCH_SIZE
                        Batch Size (default: 64)
  --num_epochs NUM_EPOCHS
                        Number of training epochs (default: 100)
  --evaluate_every EVALUATE_EVERY
                        Evaluate model on dev set after this many steps
                        (default: 100)
  --checkpoint_every CHECKPOINT_EVERY
                        Save model after this many steps (default: 100)
  --allow_soft_placement ALLOW_SOFT_PLACEMENT
                        Allow device soft device placement
  --noallow_soft_placement
  --log_device_placement LOG_DEVICE_PLACEMENT
                        Log placement of ops on devices
  --nolog_device_placement

Train:

./train.py

Evaluating

./eval.py --eval_train --checkpoint_dir="./runs/1459637919/checkpoints/"

Replace the checkpoint dir with the output from the training. To use your own data, change the eval.py script to load your data.

Convolutional Neural Network for Text Classification in Tensorflow

Related tags

Overview

Requirements

Training

Evaluating

References

Owner

Denny Britz

Orchestrating Distributed Materials Acceleration Platform Tutorial

Block-wisely Supervised Neural Architecture Search with Knowledge Distillation (CVPR 2020)

To provide 100 JAX exercises over different sections structured as a course or tutorials to teach and learn for beginners, intermediates as well as experts

Parametric Contrastive Learning (ICCV2021)

The Environment I built to study Reinforcement Learning + Pokemon Showdown

Code release for "Self-Tuning for Data-Efficient Deep Learning" (ICML 2021)

Prototype for Baby Action Detection and Classification

In this project we combine techniques from neural voice cloning and musical instrument synthesis to achieve good results from as little as 16 seconds of target data.

This is a deep learning-based method to segment deep brain structures and a brain mask from T1 weighted MRI.

Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)

A Dataset for Direct Quotation Extraction and Attribution in News Articles.

ruptures: change point detection in Python

Build Low Code Automated Tensorflow, What-IF explainable models in just 3 lines of code.

MPViT:Multi-Path Vision Transformer for Dense Prediction

Torch implementation of "Enhanced Deep Residual Networks for Single Image Super-Resolution"

Accepted at ICCV-2021: Workshop on Computer Vision for Automated Medical Diagnosis (CVAMD)

Iowa Project - My second project done at General Assembly, focused on feature engineering and understanding Linear Regression as a concept

Semi-Supervised Graph Prototypical Networks for Hyperspectral Image Classification, IGARSS, 2021.

Hardware accelerated, batchable and differentiable optimizers in JAX.

A Learning-based Camera Calibration Toolbox