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
|
tree_extension_metadata = {
'_tree':
{'sources': [cython_gen_cpp.process('_tree.pyx')],
'override_options': ['optimization=3']},
'_splitter':
{'sources': [cython_gen.process('_splitter.pyx')],
'override_options': ['optimization=3']},
'_partitioner':
{'sources': [cython_gen.process('_partitioner.pyx')],
'override_options': ['optimization=3']},
'_criterion':
{'sources': [cython_gen.process('_criterion.pyx')],
'override_options': ['optimization=3']},
'_utils':
{'sources': [cython_gen.process('_utils.pyx')],
'override_options': ['optimization=3']},
}
foreach ext_name, ext_dict : tree_extension_metadata
py.extension_module(
ext_name,
[ext_dict.get('sources'), utils_cython_tree],
dependencies: [np_dep],
override_options : ext_dict.get('override_options', []),
subdir: 'sklearn/tree',
install: true
)
endforeach
|
