Getting Started
User Guide
embedata
n_neighbors
API Reference
PRCutGradLoss
PRCutGradLoss.forward()
PRCutBatchLoss
PRCutBatchLoss.__init__()
PRCutBatchLoss.update_cluster_p()
PRCutBatchLoss.cluster_likelihood
PRCutBatchLoss.forward()
SimplexL2Loss
SimplexL2Loss.forward()
HyCutLoss
HyCutLoss.__init__()
HyCutLoss.forward()
RatioCutLoss
RatioCutLoss.__init__()
RatioCutLoss.forward()
NCutLoss
NCutLoss.__init__()
NCutLoss.compute_phi()
NCutLoss.forward()
equal_size_bins()
log_kmeans_bins()
compute_ncut_bin_phi()
Hyp2F1
Hyp2F1.forward()
Hyp2F1.backward()
build_rbf_knn_graph()
knn_graph()
gaussian_rbf_kernel()
symmetrize()
get_knn_distances()
sp_knn_similarity()
torch_knn_similarity()
torch_pairwise_similarities()
compute_sp_similarities()
sparse_laplacian()
evaluate_clustering()
nmi_score()
ari_score()
cluster_acc_score()
assign_clusters()
assign_unique_clusters()
compute_rcut_ncut()
ratio_cut_score()
soft_ncut()
soft_rcut()
ShuffledRangeDataset
ShuffledRangeDataset.__init__()
ShuffledRangeDataset.shuffle()
GradientMixer
GradientMixer.__init__()
GradientMixer.attach()
GradientMixer.detach()
GradientMixer.make_grad_scaler_hook()
get_pairs_unique_map()