From 5ac02bfda4b3ff0700344a1b8757ec3d586ea7a5 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Thu, 12 Dec 2019 19:20:24 -0700
Subject: Reformat test size

---
 main.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'main.py')

diff --git a/main.py b/main.py
index c3da14bed..88734d57f 100644
--- a/main.py
+++ b/main.py
@@ -54,8 +54,18 @@ cm_data = c_matrix_multiple(molecules, nuclear_charge, as_eig=True)
 ljm_data = lj_matrix_multiple(molecules, nuclear_charge, as_eig=True)
 
 # ML calculation.
-do_ml(cm_data, energy_pbe0, 1000, 100, sigma=1000.0, desc_type='CM')
-do_ml(ljm_data, energy_pbe0, 1000, 100, sigma=1000.0, desc_type='L-JM')
+do_ml(cm_data,
+      energy_pbe0,
+      1000,
+      test_size=100,
+      sigma=1000.0,
+      desc_type='CM')
+do_ml(ljm_data,
+      energy_pbe0,
+      1000,
+      test_size=100,
+      sigma=1000.0,
+      desc_type='L-JM')
 
 # End of program
 end_time = time.perf_counter()
-- 
cgit v1.2.3-54-g00ecf