@@ -34,7 +34,7 @@ def default_classification_model(input_width,num_classes):
3434
3535
3636class KerasClassifier (BasePandasEstimator ,BaseEstimator ,ClassifierMixin ):
37- def __init__ (self ,model_create = default_classification_model ,tmp_model = "/tmp/model" ,target = None , target_readable = None ,included = None ,excluded = None ,id_map = {},optimizer = 'adam' , loss = 'categorical_crossentropy' , train_batch_size = 128 , test_batch_size = 128 , nb_epoch = 100 , shuffle = True , show_accuracy = False , validation_split = 0 , validation_data = None , callbacks = None ,verbose = 0 ):
37+ def __init__ (self ,model_create = default_classification_model ,tmp_model = "/tmp/model" ,target = None , target_readable = None ,included = None ,excluded = None ,id_map = {},optimizer = 'adam' , loss = 'categorical_crossentropy' , train_batch_size = 128 , test_batch_size = 128 , nb_epoch = 100 , shuffle = True , validation_split = 0 , validation_data = None , callbacks = None ,verbose = 0 ):
3838 """
3939 Wrapper for keras with pandas support
4040 Derived from https://github.com/fchollet/keras/blob/master/keras/wrappers/scikit_learn.py
@@ -64,8 +64,6 @@ def __init__(self,model_create=default_classification_model,tmp_model="/tmp/mode
6464 Number of training epochs.
6565 shuffle : boolean, optional
6666 Wheter to shuffle the samples at each epoch.
67- show_accuracy : boolean, optional
68- Whether to display class accuracy in the logs at each epoch.
6967 validation_split : float [0, 1], optional
7068 Fraction of the data to use as held-out validation data.
7169 validation_data : tuple (X, y), optional
@@ -94,7 +92,6 @@ def __init__(self,model_create=default_classification_model,tmp_model="/tmp/mode
9492 self .test_batch_size = test_batch_size
9593 self .nb_epoch = nb_epoch
9694 self .shuffle = shuffle
97- self .show_accuracy = show_accuracy
9895 self .validation_split = validation_split
9996 self .validation_data = validation_data
10097 self .callbacks = [] if callbacks is None else callbacks
@@ -165,7 +162,7 @@ class labels
165162 self .compiled_model_ .compile (optimizer = self .optimizer , loss = self .loss )
166163 history = self .compiled_model_ .fit (
167164 X , y , batch_size = self .train_batch_size , nb_epoch = self .nb_epoch , verbose = self .verbose ,
168- shuffle = self .shuffle , show_accuracy = self . show_accuracy ,
165+ shuffle = self .shuffle ,
169166 validation_split = self .validation_split , validation_data = self .validation_data ,
170167 callbacks = self .callbacks )
171168
0 commit comments