Fix json serialize error

json does not recognize NumPy data types. (in this case, int64) Convert the number to a Python int before serializing the object.
kakao · Oct 13, 2020 · d8af029 · d8af029
1 parent 8beb7ba
commit d8af029
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/example_als.py b/examples/example_als.py
@@ -58,7 +58,7 @@ def example1():
     best_parameters = optimization_res['best_parameters']
 
     print(json.dumps(optimization_res['best'], indent=2))
-    print(json.dumps({'alpha': best_parameters['alpha'], 'd': best_parameters['d'],
+    print(json.dumps({'alpha': int(best_parameters['alpha']), 'd': int(best_parameters['d']),
                       'reg_u': best_parameters['reg_u'], 'reg_i': best_parameters['reg_i']}, indent=2))