Shave off another 4 seconds

cov-lineages · Feb 5, 2021 · 780891b · 780891b
1 parent 7854de9
commit 780891b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pangolin/scripts/pangolearn.py b/pangolin/scripts/pangolearn.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 
 import pandas as pd
+import numpy as np
 from sklearn.model_selection import train_test_split
 from sklearn.linear_model import LogisticRegression
 from sklearn import metrics
@@ -137,7 +138,8 @@ def readInAndFormatData(sequencesFile, indiciesToKeep, blockSize=1000):
 	idList.append(referenceId)
 
 	# create a data from from the seqList
-	df = pd.DataFrame(rows, columns=columns).astype('uint8')
+	d = np.array(rows, np.uint8)
+	df = pd.DataFrame(d, columns=columns)
 
 	predictions = loaded_model.predict_proba(df)