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 repository contains the code and models necessary to replicate the results of paper: How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspective

CSD: Consistency-based Semi-supervised learning for object Detection

Machine learning Bot detection technique, based on United States election dataset

A project to build an AI voice assistant using Python . The Voice assistant interacts with the humans to perform basic tasks.

Source code for our paper "Empathetic Response Generation with State Management"

Implementation of the master's thesis "Temporal copying and local hallucination for video inpainting".

An Abstract Cyber Security Simulation and Markov Game for OpenAI Gym

This repository is to support contributions for tools for the Project CodeNet dataset hosted in DAX

Code for our paper at ECCV 2020: Post-Training Piecewise Linear Quantization for Deep Neural Networks

A TensorFlow implementation of Neural Program Synthesis from Diverse Demonstration Videos

A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.

Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."

Object Database for Super Mario Galaxy 1/2.

A small library for doing fluid simulation with neural networks.

FasterAI: A library to make smaller and faster models with FastAI.

Source code related to the article submitted to the International Conference on Computational Science ICCS 2022 in London

Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"

[NeurIPS 2020] Official Implementation: "SMYRF: Efficient Attention using Asymmetric Clustering".

Compare outputs between layers written in Tensorflow and layers written in Pytorch

Frequency Spectrum Augmentation Consistency for Domain Adaptive Object Detection