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

Using Convolutional Neural Networks (CNN) for Semantic Segmentation of Breast Cancer Lesions (BRCA)

PyTorch code for SENTRY: Selective Entropy Optimization via Committee Consistency for Unsupervised DA

Official implementation of "Motif-based Graph Self-Supervised Learning forMolecular Property Prediction"

Container : Context Aggregation Network

Este conversor criará a medida exata para sua receita de capuccino gelado da grandiosa Rafaella Ballerini!

Instance-wise Feature Importance in Time (FIT)

MemStream: Memory-Based Anomaly Detection in Multi-Aspect Streams with Concept Drift

Code for "Hierarchical Skills for Efficient Exploration" HSD-3 Algorithm and Baselines

Job-Recommend-Competition - Vectorwise Interpretable Attentions for Multimodal Tabular Data

Software for Multimodalty 2D+3D Facial Expression Recognition (FER) UI

Code for EMNLP'21 paper "Types of Out-of-Distribution Texts and How to Detect Them"

BookMyShowPC - Movie Ticket Reservation App made with Tkinter

An implementation of the AdaOPS (Adaptive Online Packing-based Search), which is an online POMDP Solver used to solve problems defined with the POMDPs.jl generative interface.

BalaGAN: Image Translation Between Imbalanced Domains via Cross-Modal Transfer

Abstractive opinion summarization system (SelSum) and the largest dataset of Amazon product summaries (AmaSum). EMNLP 2021 conference paper.

A unified framework for machine learning with time series

CVNets: A library for training computer vision networks

Joint Versus Independent Multiview Hashing for Cross-View Retrieval[J] (IEEE TCYB 2021, PyTorch Code)

A Game-Theoretic Perspective on Risk-Sensitive Reinforcement Learning

Problem-943.-ACMP - Problem 943. ACMP