@@ -218,7 +218,7 @@ def fit(self, X, y, Xt=None, yt=None,
218218 self : returns an instance of self
219219 """
220220 set_random_seed (self .random_state )
221- tf .compat .v1 .logging .set_verbosity (tf .compat .v1 .logging .ERROR )
221+ # tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
222222
223223 Xs , ys = check_arrays (X , y , accept_sparse = True )
224224 Xt , yt = self ._get_target_data (Xt , yt )
@@ -757,9 +757,9 @@ class TwoStageTrAdaBoostR2(TrAdaBoostR2):
757757 --------
758758 >>> from sklearn.linear_model import Ridge
759759 >>> from adapt.utils import make_regression_da
760- >>> from adapt.instance_based import TrAdaBoostR2
760+ >>> from adapt.instance_based import TwoStageTrAdaBoostR2
761761 >>> Xs, ys, Xt, yt = make_regression_da()
762- >>> model = TrAdaBoostR2 (Ridge(), n_estimators=10, Xt=Xt[:10], yt=yt[:10], random_state=0)
762+ >>> model = TwoStageTrAdaBoostR2 (Ridge(), n_estimators=10, Xt=Xt[:10], yt=yt[:10], random_state=0)
763763 >>> model.fit(Xs, ys)
764764 Iteration 0 - Cross-validation score: 0.2956 (0.0905)
765765 Iteration 1 - Cross-validation score: 0.2956 (0.0905)
@@ -814,7 +814,7 @@ def fit(self, X, y, Xt=None, yt=None,
814814 sample_weight_tgt = None ,
815815 ** fit_params ):
816816 set_random_seed (self .random_state )
817- tf .compat .v1 .logging .set_verbosity (tf .compat .v1 .logging .ERROR )
817+ # tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
818818
819819 Xs , ys = check_arrays (X , y , accept_sparse = True )
820820 Xt , yt = self ._get_target_data (Xt , yt )
0 commit comments