alezhibali / aps360_project

Use deep learning (LSTM) to classify news

APS360 Group 12 Project

Final report

See APS360_Final_report.pdf

Usages of different files

1. train_test_model.py

   In the train model function, all model state files and training csc files are saved at model folder. Uncomment line 83-85 and line 109-111 to enable cuda. To run it on colab, create a folder called model on the root directory. You can also change the path in get_model_path and get_csv_path if you want.

2. all_code.ipynb

   You can run this in Google colab. Uncomment some lines to enable cuda.

3. GUI_main.py

   To use graphic interface, run this file.