Skip to content

Commit

Permalink
docs: fix typos on RVFLRegressor
Browse files Browse the repository at this point in the history
  • Loading branch information
yoshoku committed Nov 6, 2023
1 parent 042b7d3 commit 0983cc0
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -55,12 +55,12 @@ def initialize(hidden_units: 128, reg_param: 100.0, scale: 1.0, random_seed: nil
#
# @param x [Numo::DFloat] (shape: [n_samples, n_features]) The training data to be used for fitting the model.
# @param y [Numo::DFloat] (shape: [n_samples, n_outputs]) The taget values to be used for fitting the model.
# @return [MLPRegressor] The learned regressor itself.
# @return [RVFLRegressor] The learned regressor itself.
def fit(x, y)
x = ::Rumale::Validation.check_convert_sample_array(x)
y = ::Rumale::Validation.check_convert_target_value_array(y)
::Rumale::Validation.check_sample_size(x, y)
raise 'RBFRegressor#fit requires Numo::Linalg but that is not loaded.' unless enable_linalg?(warning: false)
raise 'RVFLRegressor#fit requires Numo::Linalg but that is not loaded.' unless enable_linalg?(warning: false)

y = y.expand_dims(1) if y.ndim == 1

Expand Down

0 comments on commit 0983cc0

Please sign in to comment.