Update src/bin/target_family_classifier.py

src/bin/target_family_classifier.py

@@ -100,7 +100,7 @@ def score_protein_rep(dataset):
 
     print('Producing protein family predictions...\n')
     for i in tqdm(range(10)):
-        clf = linear_model.SGDClassifier(class_weight="balanced", loss="log", penalty="elasticnet", max_iter=1000, tol=1e-3, random_state=i, n_jobs=-1)
+        clf = linear_model.SGDClassifier(class_weight="balanced", loss="log_loss", penalty="elasticnet", max_iter=1000, tol=1e-3, random_state=i, n_jobs=-1)
         clf2 = OneVsRestClassifier(clf, n_jobs=-1)
 
         train_indexx = train_index.iloc[i].astype(int)