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

This is the official code for the paper "Tracker Meets Night: A Transformer Enhancer for UAV Tracking".

CLIPort: What and Where Pathways for Robotic Manipulation

Towhee is a flexible machine learning framework currently focused on computing deep learning embeddings over unstructured data.

Learning Representational Invariances for Data-Efficient Action Recognition

Histology images query (unsupervised)

Banglore House Prediction Using Flask Server (Python)

Accelerating BERT Inference for Sequence Labeling via Early-Exit

PointNetVLAD: Deep Point Cloud Based Retrieval for Large-Scale Place Recognition, CVPR 2018

Pyramid Scene Parsing Network, CVPR2017.

GUI for a Vocal Remover that uses Deep Neural Networks.

Studying Python release adoptions by looking at PyPI downloads

A tiny, pedagogical neural network library with a pytorch-like API.

A distributed deep learning framework that supports flexible parallelization strategies.

With this package, you can generate mixed-integer linear programming (MIP) models of trained artificial neural networks (ANNs) using the rectified linear unit (ReLU) activation function

MAT: Mask-Aware Transformer for Large Hole Image Inpainting

IhoneyBakFileScan Modify - 批量网站备份文件扫描器，增加文件规则，优化内存占用

Optimal Adaptive Allocation using Deep Reinforcement Learning in a Dose-Response Study

wmctrl ported to Python Ctypes

U-Net for GBM

Pmapper is a super-resolution and deconvolution toolkit for python 3.6+