Scientists use AI to predict what makes a successful relationship