Machine learning could aid mental health diagnoses