Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
MPBA
INF
Commits
0492a38e
Commit
0492a38e
authored
Nov 28, 2019
by
Marco Chierici
Browse files
Dropped unused imports; minor improvements
parent
a128b05f
Changes
1
Hide whitespace changes
Inline
Side-by-side
sklearn_rf_training_fixrank.py
View file @
0492a38e
...
...
@@ -8,25 +8,24 @@ import numpy as np
import
pandas
as
pd
import
csv
import
os.path
from
scaling
import
norm_l2
from
mlpy
import
borda_count
,
canberra_stability
from
input_output
import
load_data
import
performance
as
perf
import
sys
import
tarfile
import
glob
import
argparse
import
configparser
as
ConfigParser
from
sklearn.ensemble
import
RandomForestClassifier
from
sklearn.multiclass
import
OneVsRestClassifier
from
sklearn.metrics
import
roc_auc_score
,
matthews_corrcoef
,
accuracy_score
from
sklearn
import
preprocessing
from
sklearn.model_selection
import
StratifiedKFold
,
StratifiedShuffleSplit
,
train_test_split
import
bootstrapped.bootstrap
as
bs
import
bootstrapped.stats_functions
as
bs_stats
__author__
=
'Marco Chierici
, Alessandro Zandona
'
__version__
=
'2.
0
'
__date__
=
'
15 December
201
6
'
__author__
=
'Marco Chierici'
__version__
=
'2.
5
'
__date__
=
'
28 Nov
201
9
'
class
myArgumentParser
(
argparse
.
ArgumentParser
):
def
__init__
(
self
,
*
args
,
**
kwargs
):
...
...
@@ -68,8 +67,8 @@ RANKFEATS = args.rankFeats
relief_k
=
args
.
reliefk
BASEFILE
=
os
.
path
.
splitext
(
os
.
path
.
basename
(
DATAFILE
))[
0
]
SVM
_TYPE
=
'RandomForest'
OUTFILE
=
os
.
path
.
join
(
OUTDIR
,
'_'
.
join
([
BASEFILE
,
SVM
_TYPE
,
'RF'
if
RANK_METHOD
==
'randomForest'
else
RANK_METHOD
]))
MODEL
_TYPE
=
'RandomForest'
OUTFILE
=
os
.
path
.
join
(
OUTDIR
,
'_'
.
join
([
BASEFILE
,
MODEL
_TYPE
,
'RF'
if
RANK_METHOD
==
'randomForest'
else
RANK_METHOD
]))
# create OUTDIR if not present
try
:
...
...
@@ -82,7 +81,7 @@ except OSError:
if
RANK_METHOD
==
'ReliefF'
:
from
relief
import
ReliefF
# add ReliefF K to OUTFILE
OUTFILE
=
os
.
path
.
join
(
OUTDIR
,
'_'
.
join
([
BASEFILE
,
SVM
_TYPE
,
RANK_METHOD
+
str
(
relief_k
)]))
OUTFILE
=
os
.
path
.
join
(
OUTDIR
,
'_'
.
join
([
BASEFILE
,
MODEL
_TYPE
,
RANK_METHOD
+
str
(
relief_k
)]))
elif
RANK_METHOD
==
'tree'
:
from
sklearn.ensemble
import
ExtraTreesClassifier
elif
RANK_METHOD
==
'KBest'
:
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment