Python DeepmdData.get_batch示例

示例#1

文件： test_deepmd_data.py 项目： Ericwang6/deepmd-kit

 def test_get_batch(self):
     dd = DeepmdData(self.data_name)
     data = dd.get_batch(5)
     self._comp_np_mat2(np.sort(data['coord'], axis=0),
                        np.sort(self.coord_bar, axis=0))
     data = dd.get_batch(5)
     self._comp_np_mat2(np.sort(data['coord'], axis=0),
                        np.sort(self.coord, axis=0))
     data = dd.get_batch(5)
     self._comp_np_mat2(np.sort(data['coord'], axis=0),
                        np.sort(self.coord_bar, axis=0))
     data = dd.get_batch(5)
     self._comp_np_mat2(np.sort(data['coord'], axis=0),
                        np.sort(self.coord, axis=0))

示例#2

 def test_get_batch(self) :
     dd = DeepmdData(self.data_name)
     data = dd.get_batch(5)