Fixing up pre_compute_distances example.

gugarosa · Jul 7, 2020 · a4d179c · a4d179c
1 parent 61acc81
commit a4d179c
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/examples/math/pre_compute_distances.py b/examples/math/pre_compute_distances.py
@@ -9,10 +9,5 @@
 # Parsing a pre-loaded numpy array
 X, Y = p.parse_loader(txt)
 
-# Splitting data into training and testing sets
-X_train, X_test, Y_train, Y_test = s.split(
-    X, Y, percentage=0.5, random_state=1)
-
 # Creating a file of pre-computed distances
-g.pre_compute_distance(X_train, 'boat_split_distances.txt',
-                       distance='log_squared_euclidean')
+g.pre_compute_distance(X, 'boat_split_distances.txt', distance='log_squared_euclidean')