https://github.com/vansh-py04/f1-personality-prediction
Find out what MBTI personality you are and which F1 driver you are associated to!
https://github.com/vansh-py04/f1-personality-prediction
bert dash data-science deep-learning f1 formula1 nlp-machine-learning personality-predicting python
Last synced: 2 months ago
JSON representation
Find out what MBTI personality you are and which F1 driver you are associated to!
- Host: GitHub
- URL: https://github.com/vansh-py04/f1-personality-prediction
- Owner: vansh-py04
- License: mit
- Created: 2025-04-08T05:19:17.000Z (2 months ago)
- Default Branch: main
- Last Pushed: 2025-04-08T08:50:08.000Z (2 months ago)
- Last Synced: 2025-04-10T02:17:51.933Z (2 months ago)
- Topics: bert, dash, data-science, deep-learning, f1, formula1, nlp-machine-learning, personality-predicting, python
- Language: Jupyter Notebook
- Homepage:
- Size: 1.71 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# F1 Driver Personality Predictor
This project predicts a user's MBTI personality type based on a text input and matches it with an F1 driver who shares the same personality type.
It uses a BERT-based model for accurate natural language understanding and is wrapped in an interactive Dash web app UI for a smooth and stylish experience.
---
# Essential Links
Dataset : [Link](https://www.kaggle.com/datasets/datasnaek/mbti-type)
Colab Notebook : Used for training the model using GPU. [Link](https://colab.research.google.com/drive/1GOBU8qIKVqnMQ1gH_6T5tyD8li98GWnO?usp=sharing)
Deployed Project :
---
# About the Project
Model: Fine-tuned using BERT (bert-base-uncased) to classify MBTI types from text inputs.
Dataset: A large corpus of personality-labeled text was used for training [Link](https://www.kaggle.com/datasets/datasnaek/mbti-type).
Training: The model was trained on Google Colab for convenience and GPU acceleration.
Frontend: Built using Plotly Dash, allowing real-time predictions and a visual UI with images of real F1 drivers.
Model weights : model.safetensors file is very large for GITHUB. You can train your own model to acquire the file, or use the deployed link.
---
# What the Model Does
Accepts a sentence or paragraph written by a user.
Processes it through a fine-tuned BERT model.
Predicts one of 16 MBTI personality types.
Matches that personality to a real-life F1 driver.
Displays the prediction and the corresponding driver's name + image.
---
# Snippet
