EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add Python API for clustering (#1385)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-09-30 11:33:15 +08:00
committed by GitHub
parent 70568c2df7
commit b965f14cf0
26 changed files with 326 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sherpa-onnx/csrc/fast-clustering.cc
View File

@@ -16,7 +16,7 @@ class FastClustering::Impl {
explicit Impl(const FastClusteringConfig &config) : config_(config) {}
std::vector<int32_t> Cluster(float *features, int32_t num_rows,
int32_t num_cols) {
int32_t num_cols) const {
if (num_rows <= 0) {
return {};
}
@@ -77,7 +77,7 @@ FastClustering::FastClustering(const FastClusteringConfig &config)
FastClustering::~FastClustering() = default;
std::vector<int32_t> FastClustering::Cluster(float *features, int32_t num_rows,
int32_t num_cols) {
int32_t num_cols) const {
return impl_->Cluster(features, num_rows, num_cols);
}
} // namespace sherpa_onnx