File size: 1,033 Bytes
7885a28 |
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 29 30 |
cluster_extension_metadata = {
'_dbscan_inner':
{'sources': ['_dbscan_inner.pyx'], 'override_options': ['cython_language=cpp']},
'_hierarchical_fast':
{'sources': ['_hierarchical_fast.pyx', metrics_cython_tree],
'override_options': ['cython_language=cpp']},
'_k_means_common':
{'sources': ['_k_means_common.pyx'], 'dependencies': [openmp_dep]},
'_k_means_lloyd':
{'sources': ['_k_means_lloyd.pyx'], 'dependencies': [openmp_dep]},
'_k_means_elkan':
{'sources': ['_k_means_elkan.pyx'], 'dependencies': [openmp_dep]},
'_k_means_minibatch':
{'sources': ['_k_means_minibatch.pyx'], 'dependencies': [openmp_dep]},
}
foreach ext_name, ext_dict : cluster_extension_metadata
py.extension_module(
ext_name,
[ext_dict.get('sources'), utils_cython_tree],
dependencies: [np_dep] + ext_dict.get('dependencies', []),
override_options : ext_dict.get('override_options', []),
cython_args: cython_args,
subdir: 'sklearn/cluster',
install: true
)
endforeach
subdir('_hdbscan')
|