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

Repo for "Event-Stream Representation for Human Gaits Identification Using Deep Neural Networks"

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

PyTorch implementation of Asymmetric Siamese (https://arxiv.org/abs/2204.00613)

Codes for the AAAI'22 paper "TransZero: Attribute-guided Transformer for Zero-Shot Learning"

Retina blood vessel segmentation with a convolutional neural network

Tensorflow 2.x implementation of Panoramic BlitzNet for object detection and semantic segmentation on indoor panoramic images.

SubOmiEmbed: Self-supervised Representation Learning of Multi-omics Data for Cancer Type Classification

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2020

Repository for Traffic Accident Benchmark for Causality Recognition (ECCV 2020)

GemNet model in PyTorch, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)

Example for AUAV 2022 with obstacle avoidance.

利用python脚本实现微信、支付宝账单的合并，并保存到excel文件实现自动记账，可查看可视化图表。

NCVX (NonConVeX): A User-Friendly and Scalable Package for Nonconvex Optimization in Machine Learning.

SAS: Self-Augmentation Strategy for Language Model Pre-training

[ECCV 2020] Gradient-Induced Co-Saliency Detection

Simple tutorials on Pytorch DDP training

Code for the paper: Adversarial Machine Learning: Bayesian Perspectives

Dynamic Multi-scale Filters for Semantic Segmentation (DMNet ICCV'2019)

Some experiments with tennis player aging curves using Hilbert space GPs in PyMC. Only experimental for now.

ReLoss - Official implementation for paper "Relational Surrogate Loss Learning" ICLR 2022