add negative numbers to training data

BenDavidAaron · Jan 25, 2018 · c312e72 · c312e72
1 parent cc84ecd
commit c312e72
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/model-development/generate-training-data.py b/model-development/generate-training-data.py
@@ -12,8 +12,8 @@
 from tqdm import tqdm
 
 def add_two_random_ints():
-	int_1 = randint(0,999)
-	int_2 = randint(0,999)
+	int_1 = randint(-9,99)
+	int_2 = randint(-9,99)
 	return "%s,%s,%s\n" % (int_1 + int_2, int_1, int_2)
 
 number_cases = abs(int(sys.argv[1]))

diff --git a/model-development/predictor-gd.pkl b/model-development/predictor-gd.pkl
diff --git a/model-development/predictor-lr.pkl b/model-development/predictor-lr.pkl