We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 50c8d49 commit 454ba0fCopy full SHA for 454ba0f
papers/alta2020/run_experiment.py
@@ -68,7 +68,7 @@ def test_classifier(kind: str,
68
69
metrics = (['uar', 'war'] + [x + '_rec' for x in dataset.classes]
70
+ [x + '_prec' for x in dataset.classes])
71
- df = pd.DataFrame(index=pd.RangeIndex(reps, name='rep'),
+ df = pd.DataFrame(index=pd.RangeIndex(1, reps + 1, name='rep'),
72
columns=metrics + ['params'])
73
74
scoring = {'war': get_scorer('accuracy'),
@@ -79,7 +79,7 @@ def test_classifier(kind: str,
79
c + '_prec': make_scorer(precision_score, average=None, labels=[i])
80
})
81
82
- for rep in range(reps):
+ for rep in range(1, reps + 1):
83
print("Rep {}".format(rep))
84
if kind == 'svm':
85
fit_params = dict(sample_weight=sample_weight)
@@ -171,7 +171,6 @@ def main():
171
)
172
173
# Results options
174
- parser.add_argument('--name', type=str, help="The results output name.")
175
parser.add_argument('--results', type=Path, help="Results directory.")
176
177
# Cross-validation options
0 commit comments