Predict the probability of the basketball matches with Deep Neural Networks
2017 Seedcup Preliminary Contest
1.train:
python train.py --model dnn --team_data_type reduce --cuda 1 --epoch 100 --batch-size 16
2.test:
python test.py --cuda 1 --model 'dnn' --epoch 200 --team_data_type 'reduce' --model_param 'train_dnn_30.pkl'
1.train:
python train.py --model dnn --team_data_type reduce --cuda 0 --epoch 100 --batch-size 16
2.test:
python test.py --cuda 0 --model 'dnn' --epoch 200 --team_data_type 'reduce' --model_param 'train_dnn_30.pkl'
- Pytorch
- XGBoost(if you do not use train_xgboost.py, you can delete the file in project)
- Scikit-Learn
This Project is Under the MIT Licence