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

tensorrt int8 量化yolov5 4.0 onnx模型

Implementation of "Fast and Flexible Temporal Point Processes with Triangular Maps" (Oral @ NeurIPS 2020)

A Partition Filter Network for Joint Entity and Relation Extraction EMNLP 2021

An open source object detection toolbox based on PyTorch

ReAct: Out-of-distribution Detection With Rectified Activations

Source code for The Power of Many: A Physarum Swarm Steiner Tree Algorithm

The official implementation of You Only Compress Once: Towards Effective and Elastic BERT Compression via Exploit-Explore Stochastic Nature Gradient.

MNE: Magnetoencephalography (MEG) and Electroencephalography (EEG) in Python

Edge Restoration Quality Assessment

Serve TensorFlow ML models with TF-Serving and then create a Streamlit UI to use them

[NeurIPS2021] Exploring Architectural Ingredients of Adversarially Robust Deep Neural Networks

CharacterGAN: Few-Shot Keypoint Character Animation and Reposing

Music Generation using Neural Networks Streamlit App

C3DPO - Canonical 3D Pose Networks for Non-rigid Structure From Motion.

Melanoma Skin Cancer Detection using Convolutional Neural Networks and Transfer Learning🕵🏻‍♂️

A simple python library for fast image generation of people who do not exist.

YOLOv5 Series Multi-backbone, Pruning and quantization Compression Tool Box.

PyTorch implementation of deep GRAph Contrastive rEpresentation learning (GRACE).

Code for classifying international patents based on the text of their titles/abstracts

EdMIPS: Rethinking Differentiable Search for Mixed-Precision Neural Networks