Abstract: Aspects of the present disclosure relate to automated data extraction and prediction using machine learning models. Embodiments include extracting, by a text extraction engine, a set of data from each document of one or more documents provided to the text extraction engine. Embodiments include instructing a machine learning model, via a prompt, to identify and classify one or more protected entities contained in the set of data from each document by parsing the set of data from each document. Embodiments include instructing the machine learning model, via the prompt, to generate, for each protected entity of the one or more protected entities identified in the set of data from each document, a corresponding unprotected entity and to replace each protected entity with the corresponding unprotected entity. Embodiments include receiving an output from the machine learning model in response to the prompt and performing an action based on the output.