moreo
/
QuaPy
1
0
Fork 1
Code Issues Pull Requests Releases Wiki Activity

switching

This commit is contained in:
Alejandro Moreo Fernandez 2024-05-15 12:00:00 +02:00
parent 2ac48a9798
commit 0df44c13a9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Retrieval/classifier_kfcv_accuracy.py
View File

@ -30,7 +30,6 @@ data_home = 'data'
datasets = ['continent', 'gender', 'years_category', 'relative_pageviews_category', 'num_sitelinks_category']
param_grid = {'C': np.logspace(-4, 4, 9), 'class_weight': ['balanced', None]}
# param_grid = {'C': np.logspace(-1, 1, 2)}
classifiers = [
('LR', LogisticRegression(max_iter=5000), param_grid),
@ -44,6 +43,9 @@ table = Table(name=f'accuracy', benchmarks=[benchmark_name(d) for d in datasets]
table.format.show_std = False
table.format.stat_test = None
table.format.lower_is_better = False
table.format.color = False
table.format.remove_zero = True
table.format.style = 'rules'
for class_name, (cls_name, cls, grid) in itertools.product(datasets, classifiers):