| 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') |
|
|
