@@ -22,10 +22,9 @@ class NiaRbfRegressor(BaseNiaRbf, RegressorMixin):
2222 + use non-linear Gaussian function with `sigmas` as standard deviation
2323 + set up regulation term with hyperparameter `regularization`
2424
25- Inherits
26- --------
27- BaseNiaRbf : The base class for NIA-based RBF networks.
28- RegressorMixin : Scikit-learn mixin class for regression estimators.
25+ Inherits:
26+ + BaseNiaRbf : The base class for NIA-based RBF networks.
27+ + RegressorMixin : Scikit-learn mixin class for regression estimators.
2928
3029 Parameters
3130 ----------
@@ -88,8 +87,8 @@ class NiaRbfRegressor(BaseNiaRbf, RegressorMixin):
8887 >>> data = Data(X, y)
8988 >>> data.split_train_test(test_size=0.2, random_state=1)
9089 >>> opt_paras = {"name": "GA", "epoch": 10, "pop_size": 30}
91- >>> model = NiaRbfRegressor(size_hidden=10, center_finder="kmeans", regularization=False, lamda=0.01,
92- >>> obj_name=None, optimizer ="BaseGA", optimizer_paras =opt_paras, verbose=True, seed=42, obj_weights=None)
90+ >>> model = NiaRbfRegressor(size_hidden=10, center_finder="kmeans", regularization=False,
91+ >>> obj_name=None, optim ="BaseGA", optim_paras =opt_paras, verbose=True, seed=42, obj_weights=None)
9392 >>> model.fit(data.X_train, data.y_train)
9493 >>> pred = model.predict(data.X_test)
9594 >>> print(pred)
0 commit comments