FIX partial fit test for MLP