🎨 Fix slightly NCA

yoshoku · Mar 1, 2020 · d485140 · d485140
1 parent 6099cd6
commit d485140
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/rumale/metric_learning/neighbourhood_component_analysis.rb b/lib/rumale/metric_learning/neighbourhood_component_analysis.rb
@@ -120,7 +120,7 @@ def optimize_components(x, y, n_features, n_components)
         dold = 0.0
         optimizer.each do |prm|
           res = prm
-          puts "[NeighbourhoodComponentAnalysis] Loss after #{res[:n_iter]} epochs: #{n_samples - res[:fnc]}" if @params[:verbose]
+          puts "[NeighbourhoodComponentAnalysis] The value of objective function after #{res[:n_iter]} epochs: #{x.shape[0] - res[:fnc]}" if @params[:verbose]
           break if (fold - res[:fnc]).abs <= @params[:tol] && (dold - res[:jcb]).abs <= @params[:tol]
           fold = res[:fnc]
           dold = res[:jcb]