Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
infer.py		infer.py
infer.sh		infer.sh
model.py		model.py
requirements.txt		requirements.txt
train.sh		train.sh
train_of_textcnn.py		train_of_textcnn.py
training.py		training.py
utils.py		utils.py

README.md

TextCNN

TextCNN is a Text Classification model, our code is inspired by Pytorch-TextCNN.

Train on imdb Dataset

Run Oneflow Training script

bash train.sh

Inference

Run Oneflow Inference script

bash infer.sh