Python DeepmdData.get_batch Beispiele

Beispiel #1

Datei: test_deepmd_data.py Projekt: 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))

Beispiel #2

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