A method includes receiving a set of documents associated with data discovery. The method further includes receiving, for each document in a subset of the set of documents, an indication of relevancy or non-relevancy of the document for an issue. The method further includes modifying one or more parameters for a machine-learning model based on the indication of relevancy or non-relevancy. The method further includes outputting, for each document in the set of documents, by a machine-learning model, a prediction probability of relevancy to an issue associated with the data discovery, and a ranking of the set of documents based on the prediction probability of relevancy. The method further includes generating a user interface that includes a sampling of the documents for review by a user, where each document is associated with a predicted relevancy tag or a predicted non-relevancy tag.