Create a list of all_classes and use it in the classes parameter of both calls to labels_from_func. You’re still passing different data to the two steps but reserve room for future classes on your model (data.c should be the same in the two passes, otherwise you’ve to tweak your weights, reserving room for new classes…)