Connect four AI Model,
Made by 라건우(Geonu La) / 백진헌(Jinheon Baek) / 태영준(Yeongjun Tae)
Python (>= 3.6)
pandas
pickle
Maybe more, just use pip install if you get an error.
Rule Base
Heuristic (Minmax, Alpha-Beta Pruning)
NN_Heuristic (Minmax, Alpha-Beta Pruning using Neural Network)
main.py
Main Function to run Connect four Game
board.py
Connect four board Class
rule.py
Rule Class for Rule base approach
heuristic.py
Connect four Heuristic Class
Heuristic Class: Simple Heuristic
NN_Heuristic Class: Mixed way Heuristic (Our Thinking + NN Model)
Data Folder
Connect four dataset (UCI Dataset)
Model Folder
Connect four Machine Learning Model
MLP : Neural Network (MLP)
SVM : Support Vector Machine
Learning (learning.ipynb)
File for Connect four Machine Learning(NN / SVM) based on the jupyter notebook