-
Notifications
You must be signed in to change notification settings - Fork 0
/
optim_measurement.py
36 lines (31 loc) · 1.24 KB
/
optim_measurement.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from RandomForests.RF import doRF
from sklearn.ensemble import RandomForestClassifier
from CART.docart import doCART
from sklearn.tree import DecisionTreeClassifier
from LR.dolr import doLR
from sklearn.linear_model import LogisticRegression
from KNN.simple_doknn import doKNN
from sklearn.neighbors import KNeighborsClassifier
from NaiveBayes.NB import doNB
from sklearn.naive_bayes import GaussianNB
from PCA.doPCA import applyPCA, applyPCAWithStandardize
import pickle
import os
appDirectory='.'
PROJECTS = [ 'Mozilla', 'Openstack', 'Wikimedia','Mirantis',]
scores = ['roc_auc']
ALGORITHMS_NAME=[doCART,doRF,doKNN,doNB,doLR]
ALGORITHMS = {
doCART.__name__: DecisionTreeClassifier,
doKNN.__name__: KNeighborsClassifier,
doLR.__name__: LogisticRegression,
doRF.__name__: RandomForestClassifier,
}
for algorithm in ALGORITHMS_NAME:
for project in PROJECTS:
f = open('./Optimization_results/{}__{}_params.pckl'.format(algorithm.__name__,project), 'rb')
print('op-',project,'-',algorithm.__name__,'=>',pickle.load(f))
f.close()
f = open('./Optimization_results/{}_no_optim__{}.pckl'.format(algorithm.__name__,project), 'rb')
print('def-',project,'-',algorithm.__name__,'=>',pickle.load(f))
f.close()