from src.database import ModelOutputs if __name__ == "__main__": # instantiate data object vis_com_list = [ ModelOutputs.from_random() for i in range(3) ] # generate random predictions [vis_com.generate_random_prediction() for vis_com in vis_com_list] for vis_com in vis_com_list: print(vis_com)