averyqi115 / data_mining_homework Goto Github PK

MNIST dataset: hand-written digit classification by CNN and SVM

stock market prediction: using LSTM-model in Pytorch

train.py and test.ipynb is for CNN-based model.

details in relevant report report for CNN model.

SVM.ipynb and SVM.html is for SVM-based model.

details in relevant report report for SVM model.

benchmark.ipynb is the benchmark for stock market prediction and hyperparameters can be changed in class Args.

train.ipynb is the lstm model prediction with feature engineering and data preprocessing, whose hyperparameters can be changed in class Args.

details in relevant report report for stock market prediction.