Update metadata output

cthoyt · Jun 22, 2019 · ca507c5 · ca507c5
1 parent 4243b18
commit ca507c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/nrl/model/util.py b/src/nrl/model/util.py
@@ -50,8 +50,8 @@ def get_metadata(self):
             'package_version': get_version(),
             # TODO get losses or training info from word2vec
             'parameters': {
-                'random_walk': self.random_walk_parameters.to_json(),
-                'word2vec': self.word2vec_parameters.to_json(),
+                'random_walk': WalkerParameters.schema().dump(self.random_walk_parameters),
+                'word2vec': Word2VecParameters.schema().dump(self.word2vec_parameters),
             },
         }