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

QT Py Media Knob using rotary encoder & neopixel ring

A PyTorch Toolbox for Face Recognition

Source code of CIKM2021 Long Paper "PSSL: Self-supervised Learning for Personalized Search with Contrastive Sampling".

OpenMMLab Text Detection, Recognition and Understanding Toolbox

Pytorch implementation of DeePSiM

TensorFlow implementation of Deep Reinforcement Learning papers

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset

Parasite: a tool allowing you to compress and decompress files, to reduce their size

Generate indoor scenes with Transformers

MOpt-AFL provided by the paper "MOPT: Optimized Mutation Scheduling for Fuzzers"

Official implementation of "Learning Forward Dynamics Model and Informed Trajectory Sampler for Safe Quadruped Navigation" (RSS 2022)

Crowd-sourced Annotation of Human Motion.

Implementation of popular SOTA self-supervised learning algorithms as Fastai Callbacks.

Code release for "Detecting Twenty-thousand Classes using Image-level Supervision".

An implementation of the "Attention is all you need" paper without extra bells and whistles, or difficult syntax

BASH - Biomechanical Animated Skinned Human

A Protein-RNA Interface Predictor Based on Semantics of Sequences

Face and Body Tracking for VRM 3D models on the web.

Solution to the first stage Quiz of Hamoye internship: Introduction to Python for Machine Learning

[CVPR 2021] Anycost GANs for Interactive Image Synthesis and Editing