step_learn.py 文件源码

def my_plot(self,session,fd,y,Y,p,p1,Z_placeholder,additional_placeholders):
        max_class = tf.arg_max(p1, dimension=1)
        max_class_val = (session.run(max_class,feed_dict=fd)).astype(np.int32)
        y_val = self._Z[max_class_val]
        Y = sort_by_p(Y,p)
        y_val = sort_by_p(y_val,p)
        p = sorted(p)
        plt.plot(p,Y,'g--',p,y_val,'r')
        self.my_show()