sus
commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -105,22 +105,6 @@ Name: zzzzzz => Prediction: LABEL_0, Score: 0.9991
|
|
| 105 |
Name: uhyhu => Prediction: LABEL_0, Score: 0.9944
|
| 106 |
Name: trump => Prediction: LABEL_1, Score: 0.9998
|
| 107 |
```
|
| 108 |
-
### Training
|
| 109 |
-
|
| 110 |
-
- **Reinforcement Learning Model:**
|
| 111 |
-
Use the provided training notebook/code block to set up the custom Gym environment and train a PPO agent for name validation.
|
| 112 |
-
|
| 113 |
-
- **Transformer-based Model:**
|
| 114 |
-
Fine-tune a transformer model (e.g., DistilBERT) on a balanced dataset of real and fake names. The final model is saved in `.safetensors` format for robust, secure, and efficient storage.
|
| 115 |
-
|
| 116 |
-
### Testing
|
| 117 |
-
|
| 118 |
-
- **Confusion Matrix:**
|
| 119 |
-
Run the testing code block to evaluate the transformer-based model. The block collects predictions on a test set, computes a confusion matrix, and visualizes the results using Seaborn.
|
| 120 |
-
|
| 121 |
-
- **Flask API:**
|
| 122 |
-
Deploy a Flask API to accept a first name as input and return a prediction (real or fake) in real time.
|
| 123 |
-
|
| 124 |
## Support Me
|
| 125 |
|
| 126 |
If you find this project helpful and would like to support my work, please consider donating using crypto.
|
|
|
|
| 105 |
Name: uhyhu => Prediction: LABEL_0, Score: 0.9944
|
| 106 |
Name: trump => Prediction: LABEL_1, Score: 0.9998
|
| 107 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 108 |
## Support Me
|
| 109 |
|
| 110 |
If you find this project helpful and would like to support my work, please consider donating using crypto.
|