Currently ShapModelInterpreter plot method either plots the graph for train set or for test set. We want to make sure you can plot both in one graph