text_plain_output_1.png
import pickle

import torch

test_X = pickle.load(open('/kaggle/input/embedder/test_embedding', 'rb'))

multilayerperceptron = torch.load('/kaggle/input/nlptrain/mlp.pt', map_location=torch.device('cpu'))

test_X_tensor = torch.tensor(test_X)
multilayerperceptronprediction_proba = multilayerperceptron(test_X_tensor).detach().numpy()