Fix initialization for multi label model, add tests.