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

Code implementation of "Sparsity Probe: Analysis tool for Deep Learning Models"

A whale detector design for the Kaggle whale-detector challenge!

Applying PVT to Semantic Segmentation

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

A novel pipeline framework for multi-hop complex KGQA task. About the paper title: Improving Multi-hop Embedded Knowledge Graph Question Answering by Introducing Relational Chain Reasoning

Medical Image Segmentation using Squeeze-and-Expansion Transformers

Traditional deepdream with VQGAN+CLIP and optical flow. Ready to use in Google Colab

Snscrape-jsonl-urls-extractor - Extracts urls from jsonl produced by snscrape

This implements one of result networks from Large-scale evolution of image classifiers

Scene-Text-Detection-and-Recognition (Pytorch)

PyTorch EO aims to make Deep Learning for Earth Observation data easy and accessible to real-world cases and research alike.

Embeddinghub is a database built for machine learning embeddings.

Official PaddlePaddle implementation of Paint Transformer

Implementation of the paper ''Implicit Feature Refinement for Instance Segmentation''.

PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)

Pyramid addon for OpenAPI3 validation of requests and responses.

RIFE: Real-Time Intermediate Flow Estimation for Video Frame Interpolation

Get 2D point positions (e.g., facial landmarks) projected on 3D mesh

[ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, Zhangyang Wang

Differentiable Surface Triangulation