I don’t see any attached Rhino file? Replacing the colors generated by ColorB and ColorJ with your color palette is relatively trivial. It can literally be as simple as replacing those clusters since it is isolated from the data tree structures.
I can’t help you with machine learning.