Machine learning helps diagnose psychosis and depression