Skip to content

Commit

Permalink
add classical methods
Browse files Browse the repository at this point in the history
  • Loading branch information
shinshiner committed Jun 13, 2018
1 parent 4725188 commit ad375fe
Show file tree
Hide file tree
Showing 8 changed files with 100 additions and 0 deletions.
20 changes: 20 additions & 0 deletions traditional_methods/DecisionTree.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from sklearn.tree import DecisionTreeClassifier

import sys
sys.path.append('../')

from data_loader import MnistLoader

def DT():
loader = MnistLoader(flatten=True, data_path='../data', var_per=None)
model = DecisionTreeClassifier(criterion='gini')

model.fit(loader.data_train, loader.label_train)
print('model trained')
res = model.score(loader.data_test, loader.label_test)
print(res)

return res

if __name__ == '__main__':
DT()
20 changes: 20 additions & 0 deletions traditional_methods/KNN.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from sklearn.neighbors import KNeighborsClassifier

import sys
sys.path.append('../')

from data_loader import MnistLoader

def KNN():
loader = MnistLoader(flatten=True, data_path='../data', var_per=None)
model = KNeighborsClassifier(n_neighbors=5)

model.fit(loader.data_train, loader.label_train)
print('model trained')
res = model.score(loader.data_test, loader.label_test)
print(res)

return res

if __name__ == '__main__':
KNN()
20 changes: 20 additions & 0 deletions traditional_methods/LR.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from sklearn.linear_model import LogisticRegression

import sys
sys.path.append('../')

from data_loader import MnistLoader

def LR():
loader = MnistLoader(flatten=True, data_path='../data', var_per=None)
model = LogisticRegression(penalty='l2')

model.fit(loader.data_train, loader.label_train)
print('model trained')
res = model.score(loader.data_test, loader.label_test)
print(res)

return res

if __name__ == '__main__':
LR()
20 changes: 20 additions & 0 deletions traditional_methods/MLP.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from sklearn.neural_network import MLPClassifier

import sys
sys.path.append('../')

from data_loader import MnistLoader

def MLP():
loader = MnistLoader(flatten=True, data_path='../data', var_per=None)
model = MLPClassifier(hidden_layer_sizes=(500, 500, 500))

model.fit(loader.data_train, loader.label_train)
print('model trained')
res = model.score(loader.data_test, loader.label_test)
print(res)

return res

if __name__ == '__main__':
MLP()
Empty file added traditional_methods/SGD.py
Empty file.
20 changes: 20 additions & 0 deletions traditional_methods/SVM.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from sklearn.svm import SVC

import sys
sys.path.append('../')

from data_loader import MnistLoader

def SVM():
loader = MnistLoader(flatten=True, data_path='../data', var_per=None)
model = SVC(kernel='rbf')

model.fit(loader.data_train, loader.label_train)
print('model trained')
res = model.score(loader.data_test, loader.label_test)
print(res)

return res

if __name__ == '__main__':
SVM()
Empty file added traditional_methods/run.py
Empty file.
Empty file added traditional_methods/utils.py
Empty file.

0 comments on commit ad375fe

Please sign in to comment.