metadata
tags:
- mbti
- personality-classification
- bilstm
- pytorch
MBTI Personality Classifier
BiLSTM with Character-level CNN for Myers-Briggs Type Indicator prediction
Usage
from huggingface_hub import hf_hub_download
import pickle
import torch
# Load components
model_path = hf_hub_download(repo_id="MalekOthman/mbti_bilstm", filename="pytorch_model.bin")
config = pickle.load(open(hf_hub_download(repo_id="MalekOthman/mbti_bilstm", filename="config.pkl"), "rb"))
# Initialize model (requires AdvancedBiLSTM class definition)
model = AdvancedBiLSTM(config['word_vocab_size'], config['char_vocab_size'])
model.load_state_dict(torch.load(model_path))