Add Python API for clustering (#1385)

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

View File

@@ -12,12 +12,23 @@
namespace sherpa_onnx {
struct FastClusteringConfig {
// If greater than 0, then threshold is ignored
// If greater than 0, then threshold is ignored.
//
// We strongly recommend that you set it if you know the number of clusters
// in advance
int32_t num_clusters = -1;
// distance threshold
// distance threshold.
//
// The lower, the more clusters it will generate.
// The higher, the fewer clusters it will generate.
float threshold = 0.5;
FastClusteringConfig() = default;
FastClusteringConfig(int32_t num_clusters, float threshold)
: num_clusters(num_clusters), threshold(threshold) {}
std::string ToString() const;
void Register(ParseOptions *po);