MalekOthman commited on
Commit
b81f5c2
·
verified ·
1 Parent(s): 5535f9c

Initial model upload

Browse files

BiLSTM MBTI classifier with character-level CNN

Files changed (4) hide show
  1. README.md +26 -0
  2. char_tokenizer.pkl +3 -0
  3. model_config.pkl +3 -0
  4. word_tokenizer.pkl +3 -0
README.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - mbti
4
+ - personality-classification
5
+ - bilstm
6
+ - pytorch
7
+ ---
8
+
9
+ # MBTI Personality Classifier
10
+
11
+ BiLSTM with Character-level CNN for Myers-Briggs Type Indicator prediction
12
+
13
+ ## Usage
14
+ ```python
15
+ from huggingface_hub import hf_hub_download
16
+ import pickle
17
+ import torch
18
+
19
+ # Load components
20
+ model_path = hf_hub_download(repo_id="MalekOthman/mbti_bilstm", filename="pytorch_model.bin")
21
+ config = pickle.load(open(hf_hub_download(repo_id="MalekOthman/mbti_bilstm", filename="config.pkl"), "rb"))
22
+
23
+ # Initialize model (requires AdvancedBiLSTM class definition)
24
+ model = AdvancedBiLSTM(config['word_vocab_size'], config['char_vocab_size'])
25
+ model.load_state_dict(torch.load(model_path))
26
+ ```
char_tokenizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a872139ea1fcf23d4e69f6c2be8d0c5f22ffc96846f446b07dee7577c93015d9
3
+ size 1977
model_config.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2aaf05693e8041d34eee0ad01fd2dc3b337ce3c0231c739f90e6e6a593bd4059
3
+ size 165
word_tokenizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3dec1c26ecbb82d8f921f3b793a6d0d6e1d2b6a904318b115d58d3cf5c1e3441
3
+ size 7551069