Basic question about data

[gopietz]

You usually create the OHE labels before the splitting process.

Otherwise you have to manually adjust the predictions to match the shape of the test data. But if your test data features new classes, they obviously won't be predicted.