Added the split functionality to the colors array of EmptyEnv in negated_goals. The colors get conceptually split into train and evaluation sets by limiting the indices. The usage is akin to PyTorch style set_train() and set_eval() modes. Training mode restricts the indices to [0, len(colors) split) and evaluation mode restricts the indices to [len(colors) split, len(colors)). It is extensible to types of objects too as the split function simply takes the length of an array as the argument.
Added the split functionality to the colors array of EmptyEnv in negated_goals. The colors get conceptually split into train and evaluation sets by limiting the indices. The usage is akin to PyTorch style set_train() and set_eval() modes. Training mode restricts the indices to [0, len(colors) split) and evaluation mode restricts the indices to [len(colors) split, len(colors)). It is extensible to types of objects too as the split function simply takes the length of an array as the argument.