diff --git a/proglearn/deciders.py b/proglearn/deciders.py index f854f1a912..e8a3a5a37b 100755 --- a/proglearn/deciders.py +++ b/proglearn/deciders.py @@ -35,7 +35,7 @@ def fit( classes=None, ): if self.classes == None and len(y) == 0: - raise ValueError("Classification Decider classes undefined with no class labels fed to 'fit') + raise ValueError("Classification Decider classes undefined with no class labels fed to fit") self.classes = self.classes if self.classes != None else np.unique(y) self.transformer_id_to_transformers = transformer_id_to_transformers self.transformer_id_to_voters = transformer_id_to_voters diff --git a/proglearn/forest.py b/proglearn/forest.py index addf269470..e802653cba 100644 --- a/proglearn/forest.py +++ b/proglearn/forest.py @@ -7,7 +7,7 @@ from .voters import TreeClassificationVoter from .deciders import SimpleAverage -class L2F: +class LifelongClassificationForest: def __init__(self, n_estimators=100, tree_construction_proportion=0.67, finite_sample_correction=False): self.n_estimators = n_estimators self.tree_construction_proportion=tree_construction_proportion diff --git a/proglearn/network.py b/proglearn/network.py index d58cf93e06..df0839f03f 100644 --- a/proglearn/network.py +++ b/proglearn/network.py @@ -13,7 +13,7 @@ from keras.optimizers import Adam from keras.callbacks import EarlyStopping -class L2N: +class LifelongClassificationNetwork: def __init__( self, network,