I have build a model with tfjs based on the [IRIS Classification example][1]. After I removed the third label the AI doesn\'t work anymore and it\'s Loss and Accuracy lock a