survival prediction in Python