Research on Emotional Infection of Passengers during the SRtP of a Cruise Ship by Combining an SIR Model and Machine Learning

The Safe Return to Port issue regarding cruise ships has been extensively researched, covering aspects such as performance, operations, and electrical systems. However, an often overlooked aspect is the potential eruption of negative emotions among passengers during SRtP. This study aims to investigate the prediction of collective emotions to facilitate timely safety planning and enhance the safety of the Safe Return to Port process. To achieve this objective, an improved susceptible-infectious-recovered model with bidirectional infection is proposed to describe the emotional contagion process during the Safe Return to Port process. This model classifies the population into five emotional (extremely anxious–anxious–normal–calm–very calm) states and introduces two sources of infection. Moreover, it allows for emotions to transition both positively and negatively, making it a more realistic representation of scenarios resembling long-term refuge scenarios. In this study, questionnaire data, collected and statistically analyzed, serve as the primary dataset. A machine learning technique (the weighted random forest algorithm) is integrated with the model to make predictions. The accuracy, precision, recall, and the F-measure of prediction results demonstrate good performance. Additionally, through simulation, this study illustrates the fluctuating nature of emotional changes during the Safe Return to Port process of the cruise ship and analyzes the effects of varying parameters. The findings suggest that the improved susceptible-infectious-recovered model proposed in this paper can provide valuable insights for cruise ship emergency planning and positively contribute to maintaining passenger emotional stability during the Safe Return to Port process.