bug fix

triplet_margin_loss.py

@@ -58,17 +58,6 @@ Examples:
 
 
 _CITATION = """
-@article{scikit-learn,
-  title={Scikit-learn: Machine Learning in {P}ython},
-  author={Pedregosa, F. and Varoquaux, G. and Gramfort, A. and Michel, V.
-         and Thirion, B. and Grisel, O. and Blondel, M. and Prettenhofer, P.
-         and Weiss, R. and Dubourg, V. and Vanderplas, J. and Passos, A. and
-         Cournapeau, D. and Brucher, M. and Perrot, M. and Duchesnay, E.},
-  journal={Journal of Machine Learning Research},
-  volume={12},
-  pages={2825--2830},
-  year={2011}
-}
 @article{schultz2003learning,
   title={Learning a distance metric from relative comparisons},
   author={Schultz, Matthew and Joachims, Thorsten},
@@ -104,7 +93,7 @@ class TripletMarginLoss(evaluate.EvaluationModule):
         for a, p, n in zip(anchor, positive, negative):
             d_a_p_sum += (a - p)**2
             d_a_n_sum += (a - n)**2
-            loss = max(np.sqrt(d_a_p_sum) - np.sqrt(d_a_n_sum) + margin, 0)
+        loss = max(np.sqrt(d_a_p_sum) - np.sqrt(d_a_n_sum) + margin, 0)
         return {
             "triplet_margin_loss": float(
                 loss