A computer system can perform a semi-supervised machine learning processes to cluster a plurality of entities within a population based on their features and associated labels. The computer system can generate visualization data representing the clusters of entities and associated labels for displaying on a user interface. A user can review the clustering of entities and use the user interface to add or modify the labels associated with a particular entity or set of entities. The computer system can use the user's feedback to update the labels and then re-determine the clustering of entities using the semi-supervised machine learning process with the updated labels as input. As such, the computer system can use the user's feedback to improve the accuracy of the machine learning model without requiring a larger amount of labeled input data.