Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,22 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
# Electric Vehicle Type Classifier
|
| 4 |
+
|
| 5 |
+
## Model Description
|
| 6 |
+
A PyTorch neural network that classifies electric vehicles as Battery Electric Vehicle (BEV) or Plug-in Hybrid Electric Vehicle (PHEV) based on vehicle characteristics.
|
| 7 |
+
|
| 8 |
+
## Model Architecture
|
| 9 |
+
- Input: 9 features (make, model, year, range, price, etc.)
|
| 10 |
+
- Hidden layers: [128, 64, 32] neurons
|
| 11 |
+
- Output: Binary classification (BEV vs PHEV)
|
| 12 |
+
- Accuracy: ~XX% on test set
|
| 13 |
+
|
| 14 |
+
## Usage
|
| 15 |
+
```python
|
| 16 |
+
import torch
|
| 17 |
+
from model import TabularModel
|
| 18 |
+
|
| 19 |
+
# Load model
|
| 20 |
+
checkpoint = torch.load('ev_classifier_model.pth')
|
| 21 |
+
model = TabularModel(input_size=9, hidden_sizes=[128, 64, 32], output_size=2)
|
| 22 |
+
model.load_state_dict(checkpoint['model_state_dict'])
|