Skip to content

Commit aa39df6

Browse files
committed
added support for data id
1 parent 6824259 commit aa39df6

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

openml/evaluations/evaluation.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,14 @@
22
class OpenMLEvaluation(object):
33

44
def __init__(self, run_id, task_id, setup_id, flow_id, flow_name,
5-
data_name, function, upload_time, value, array_data=None):
5+
data_id, data_name, function, upload_time, value,
6+
array_data=None):
67
self.run_id = run_id
78
self.task_id = task_id
89
self.setup_id = setup_id
910
self.flow_id = flow_id
1011
self.flow_name = flow_name
12+
self.data_id = data_id
1113
self.data_name = data_name
1214
self.function = function
1315
self.upload_time = upload_time

openml/evaluations/functions.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,10 @@ def _list_evaluations(api_call):
8484

8585
evaluation = OpenMLEvaluation(int(eval_['oml:run_id']), int(eval_['oml:task_id']),
8686
int(eval_['oml:setup_id']), int(eval_['oml:flow_id']),
87-
eval_['oml:flow_name'], eval_['oml:data_name'],
88-
eval_['oml:function'], eval_['oml:upload_time'],
89-
float(eval_['oml:value']), array_data)
87+
eval_['oml:flow_name'], eval_['oml:data_id'],
88+
eval_['oml:data_name'], eval_['oml:function'],
89+
eval_['oml:upload_time'], float(eval_['oml:value']),
90+
array_data)
9091
evals[run_id] = evaluation
9192
return evals
9293

0 commit comments

Comments
 (0)