Update prediction.py

biolib · Jan 30, 2019 · 68790ae · 68790ae
1 parent 833d201
commit 68790ae
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/prediction.py b/prediction.py
@@ -17,11 +17,9 @@
 
 predicted_dihedral_angles, predicted_backbone_atoms, batch_sizes = model(input_senquences_encoded)
 
-predicted_dihedral_angles, _ = calculate_dihedral_angles_over_minibatch(predicted_backbone_atoms, batch_sizes, False)
-
 write_to_pdb(
     get_structure_from_angles(input_senquences_encoded[0], predicted_dihedral_angles[:,0]),
     "myprediction"
 )
 
-print("Wrote prediction to output/protein_myprediction.pdb")
+print("Wrote prediction to output/protein_myprediction.pdb")