Thanks. So the train_mask and test_mask looks like this? (if I have no validation set) train_mask = [ [index1, feature1], [index2, feature2], … , [indexN, featureN] ] and test_mask = [ [indexN+1, featureN+1], [indexN+2, featureN+2], … , [indexM, featureM] ] ?