Download and put the following datasets in the data
dir.
Run the main.py
file with the following arguments:
--model
Model Type (transformer
orlstm
)--dataset
Dataset to use (news
orsentiment
)--batchsize
(an integer)--epoch
(an integer)--normalize
to normalize the inputs (0
(False) or1
(True))--trans
to transliterate the input (0
(False) or1
(True))
python main.py --model lstm --dataset news