lib/umappp.rb in umappp-0.1.4 vs lib/umappp.rb in umappp-0.1.5

@@ -35,9 +35,10 @@
   # @param negative_sample_rate [Numeric]
   # @param num_neighbors [Integer]
   # @param seed [Integer]
   # @param batch [Boolean]
   # @param num_threads [Integer]
+  # @return [Numo::SFloat] the final embedding
 
   def self.run(embedding, method: :annoy, ndim: 2, tick: 0, **params)
     unless (u = (params.keys - default_parameters.keys)).empty?
       raise ArgumentError, "[umappp.rb] unknown option : #{u.inspect}"
     end