X_train_dict = pandas.DataFrame(X_train[:, 1:]).T.to_dict().values() X_test_dict = pandas.DataFrame(X_test[:, 1:]).T.to_dict().values() # We create a pipeline.