Patent ID: 11860977
Assignee: AMAZON TECHNOLOGIES, INC.
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G

Claim 3:
4. A computer-implemented method comprising:
generating a k-nearest neighbors graph comprising a first plurality of nodes connected by a first plurality of edges for a set of features from a dataset where k is a positive integer;
generating, by a machine learning model for an input of the k-nearest neighbors graph, a linkage probability for each of the first plurality of edges that indicates a probability that an edge is linking nodes that have a same class;
generating, by the machine learning model, a density probability for each of the first plurality of nodes that indicates a probability that a node is a center of a class of its k-nearest neighbors;
merging the first plurality of nodes into a second plurality of nodes, having fewer nodes than the first plurality of nodes, connected by a second plurality of edges based at least in part on the linkage probabilities and the density probabilities; and
transmitting information that identifies the second plurality of nodes.