A method including obtaining input attribute values and a target risk attribute value associated with a first borehole segment. The method also includes training a prediction model for the target risk attribute using the input attribute values and the target risk attribute value. The method also includes acquiring subsequent input attribute values. The method also includes using the trained prediction model and the subsequent input attribute values to predict a target risk attribute value for a second borehole segment. The method also includes storing or displaying information based on the predicted target risk attribute value.