GitHub - mikelee1/text

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
.idea		.idea
__pycache__		__pycache__
checkpoints		checkpoints
data		data
helper		helper
images		images
tensorboard		tensorboard
testfile		testfile
README		README
cnn_model.py		cnn_model.py
cnn_model.pyc		cnn_model.pyc
filter.py		filter.py
predict.py		predict.py
rnn_model.py		rnn_model.py
run_cnn.py		run_cnn.py
run_rnn.py		run_rnn.py
source.csv		source.csv
total.csv		total.csv

Repository files navigation

1. train the model:
   python run_cnn.py train or python run_rnn.py train

2. predict the album class:
   python predict.py inter # interact style
   python predict.py file sourcefilename desfilename #source file content should be like this: 
   XXX 
   XXX 
   XX 
   XXXXX