I am trying to classify layouts of a factory whether it is recommended or not based upon the image of the layout along with the reward (int variable) provided to it. Is it possi