A system and method are disclosed for training a machine learning model using information pertaining to transmissions of one or more media items to user devices associated with a user account. Generating training data for the machine learning model includes generating first contextual information associated with a first user device and generating a first target output that identifies an indication of a preference of a user preference to cancel the first transmission. The method includes providing the training data to train the machine learning model.