faisalsns commited on
Commit
7f1b875
·
verified ·
1 Parent(s): 7bb1e78

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -3
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'])