Pranaam uses the Bihar Land Records data, plot-level land records (N= 41.87 million plots or 12.13 individuals/accounts across 35,626 villages), to build machine learning models that predict religion and caste from the name. Our final dataset has around 4M unique records. To learn how to transform the data and the models underlying the package, check the notebooks.
The first function we are releasing with the package is [pred_rel]{.title-ref}, which predicts religion based on the name (currently only [muslim]{.title-ref} or [not]{.title-ref}). (For context, nearly 95% of India's population are Hindu or Muslim, with Sikhs, Buddhists, Christians, and other groups making up the rest.) The OOS accuracy assessed on unseen names is nearly 98% for both Hindi and English models.
Our training data is in Hindi. To build models that classify names provided in English, we used the indicate package to transliterate our training data to English.
We are releasing this software in the hope that it enables activists and researchers
- Highlight biases
- Fight biases
- Prevent biases (regress out some of these biases in models built on natural language corpus with person names).
We strongly recommend installing [pranaam]{.title-ref} inside a Python virtual environment. (see venv documentation)
pip install pranaam
- pranaam.pred_rel takes a list of Hindi/English names and predicts whether the person is Muslim or not.
By using names in English :
from pranaam import pranaam
names = ["Shah Rukh Khan", "Amitabh Bachchan"]
result = pranaam.pred_rel(names)
print(result)
output -:
name pred_label pred_prob_muslim
0 Shah Rukh Khan muslim 73.0
1 Amitabh Bachchan not-muslim 27.0
By using names in Hindi :
from pranaam import pranaam
names = ["शाहरुख खान", "अमिताभ बच्चन"]
result = pranaam.pred_rel(names, lang="hin")
print(result)
output -:
name pred_label pred_prob_muslim
0 शाहरुख खान muslim 73.0
1 अमिताभ बच्चन not-muslim 27.0
We expose one function, which takes Hindi/English text (name) and predicts religion and caste.
- pranaam.pred_rel(input)
- What it does:
- predicts religion based on hindi/english text (name)
- Output
- Returns pandas with name and label (muslim/not-muslim)
- What it does:
Rajashekar Chintalapati, Aaditya Dar, and Gaurav Sood
- appeler/naampy — Infer Sociodemographic Characteristics from Names Using Indian Electoral Rolls
- appeler/namesexdata — Data on international first names and sex of people with that name
- appeler/naamkaran — generative model for names
- appeler/outkast — Using data from over 140M+ Indians from the SECC 2011, we map last names to caste (SC, ST, Other)
- appeler/parsernaam — AI name parsing. Predict first or last name using a DL model.
The project welcomes contributions from everyone! It depends on it. To maintain this welcoming atmosphere and to collaborate in a fun and productive way, we expect contributors to the project to abide by the Contributor Code of Conduct.
The package is released under the MIT License.