def __init__()

in submission_code/best_util.py [0:0]

• 9 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, config, file_path, model_path, device):
        self.config = config
        self.device = device
        
        self.cmd_le = joblib.load(f'{file_path}/cmd_encoder')
        self.text_tokenizer = spm.SentencePieceProcessor(f'{file_path}/txt_bpe_clf.model')
        
        self.model = BertClassifier(self.config, pad_id, len(self.cmd_le.classes_))
        self.model.load_state_dict(torch.load(model_path, map_location=device)['model_state_dict'])
        self.model.eval()
        self.model.to(device)