提交 962939d3 编辑于 作者: Toshihiro Nakae's avatar Toshihiro Nakae
浏览文件

add error handlings

上级 cebf5384
加载中
加载中
加载中
加载中
+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)