feat: add 0 into the train dataset
feat: add 1 into the train dataset