PREDICTION DEVICE FOR PREDICTING INFORMATION RELATED TO PATIENT, OPERATION METHOD OF PREDICTION DEVICE, AND PROGRAM

A prediction device that predicts information related to a patient based on medical data of the patient, including: a processor; and a memory connected the processor, in which the processor being configured to execute: data set extraction processing of extracting M data sets by classifying pieces of medical data of plural patients into any one of M types of predetermined attributes, training data set generation processing of generating M training data sets related to the M types of attributes from the M data sets, similarity calculation processing of calculating, for each pair of the M types of attributes, a similarity between the attributes, training processing of training one or plural machine learning models based on the similarity between the attributes by using the M training data sets, and prediction processing of causing the one or plural machine learning models to predict the information related to the patient.