MalekOthman/mbti_bilstm_personality_prediction

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))