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

[MICCAI'20] AlignShift: Bridging the Gap of Imaging Thickness in 3D Anisotropic Volumes

Tutorials, assignments, and competitions for MIT Deep Learning related courses.

2D Time independent Schrodinger equation solver for arbitrary shape of well

Implementation of the paper titled "Using Sampling to Estimate and Improve Performance of Automated Scoring Systems with Guarantees"

Imaginaire - NVIDIA's Deep Imagination Team's PyTorch Library

This program presents convolutional kernel density estimation, a method used to detect intercritical epilpetic spikes (IEDs)

K-PLUG: Knowledge-injected Pre-trained Language Model for Natural Language Understanding and Generation in E-Commerce (EMNLP Founding 2021)

Real-time Joint Semantic Reasoning for Autonomous Driving

Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners

VITS: Conditional Variational Autoencoder with Adversarial Learning for End-to-End Text-to-Speech

All the code and files related to the MI-Lab of UE19CS305 course in sem 5

[ICCV'2021] Image Inpainting via Conditional Texture and Structure Dual Generation

DLWP: Deep Learning Weather Prediction

No-Reference Image Quality Assessment via Transformers, Relative Ranking, and Self-Consistency

Hierarchical Few-Shot Generative Models

QMagFace: Simple and Accurate Quality-Aware Face Recognition

Doge-Prediction - Coding Club prediction ig

MoveNet Single Pose on OpenVINO

Relative Uncertainty Learning for Facial Expression Recognition

A annotation of yolov5-5.0