Negative binomial regression - Friends after marriage - Python