def populateAxes(self, ax, columns=3): """ Returns the legend as a matplotlib artist Arguments: - columns: the number of columns of feature / representation pairs """ ax.set_xlim(0, 600) ax.set_ylim(-800, 50) result = [] x = y = 0 for track in self.policy()._makeTrackDefns(): if track.tag is None or track.tag == "Graphs": continue ax.text(10, y * 30, track.tag) y -= 1 for feature in track: seq = self._makeSampleSequence(feature) display = Display( seq, policy=self.policy, min_feature_height=10, show_code=False, pad=0, ) sample = display.makeArtist() #trans = sample.get_transform() #offset = Affine2D() #offset.translate(x*600+20 / columns, y*30) sample.translate(x * 600 / columns + 10, y * 30) ax.add_artist(sample) ax.text(x * 600 / columns + 90, y * 30, feature) x += 1 if x % columns == 0: x = 0 y -= 1 if x: x = 0 y -= 1 ax.axhline((y + .7) * 30)
def populateAxes(self, ax, columns = 3): """ Returns the legend as a matplotlib artist Arguments: - columns: the number of columns of feature / representation pairs """ ax.set_xlim(0, 600) ax.set_ylim(-800, 50) result = [] x = y = 0 for track in self.policy()._makeTrackDefns(): if track.tag is None or track.tag=="Graphs": continue ax.text(10, y*30, track.tag) y -= 1 for feature in track: seq = self._makeSampleSequence(feature) display = Display(seq, policy = self.policy, min_feature_height = 10, show_code = False, pad = 0,) sample = display.makeArtist() #trans = sample.get_transform() #offset = Affine2D() #offset.translate(x*600+20 / columns, y*30) sample.translate(x*600/columns+10, y*30) ax.add_artist(sample) ax.text(x*600/columns+90, y*30, feature) x += 1 if x % columns == 0: x = 0 y -= 1 if x: x = 0 y -= 1 ax.axhline((y+.7)*30)