加载中 dagmm/dagmm.py +6 −0 原始行号 差异行号 差异行 加载中 @@ -156,10 +156,16 @@ class DAGMM: energies : array-like, shape (n_samples) Calculated sample energies. """ if self.sess is None: raise Exception("Trained model does not exist.") energies = self.sess.run(self.energy, feed_dict={self.input:x}) return energies def save(self, fdir): if self.sess is None: raise Exception("Trained model does not exist.") if not exists(fdir): makedirs(fdir) 加载中 加载中
dagmm/dagmm.py +6 −0 原始行号 差异行号 差异行 加载中 @@ -156,10 +156,16 @@ class DAGMM: energies : array-like, shape (n_samples) Calculated sample energies. """ if self.sess is None: raise Exception("Trained model does not exist.") energies = self.sess.run(self.energy, feed_dict={self.input:x}) return energies def save(self, fdir): if self.sess is None: raise Exception("Trained model does not exist.") if not exists(fdir): makedirs(fdir) 加载中