Skip to content

Commit e607a33

Browse files
committed
remove keras_iris from build and update keras version
1 parent f564c71 commit e607a33

File tree

5 files changed

+7
-8
lines changed

5 files changed

+7
-8
lines changed

build/Makefile

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ ALL_LIST=\
1515
seldonio_examples_ml10m \
1616
seldonio_examples_reuters_data \
1717
seldonio_reuters_example \
18-
seldonio_iris_keras \
1918
seldonio_mysql \
2019
seldonio_grafana \
2120
seldonio_iago \

docker/examples/iris/keras/Dockerfile.in

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
FROM seldonio/pyseldon:%SELDON_PYTHON_PACKAGE_VERSION%
22

3+
RUN pip install theano
4+
35
ENV SELDON_HOME=/home/seldon
6+
ENV KERAS_BACKEND=theano
47
COPY keras_pipeline.py $SELDON_HOME/keras_pipeline.py
58
COPY create-json.py $SELDON_HOME/create-json.py
69

docker/pyseldon/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ RUN pip install kazoo filechunkio Flask gunicorn pylibmc gensim annoy smart_open
4343

4444
RUN cd /usr/local/src && mkdir keras && cd keras && \
4545
git clone https://github.com/fchollet/keras.git && \
46-
cd keras && git checkout tags/0.2.0 && python setup.py install
46+
cd keras && git checkout tags/2.0.0 && python setup.py install
4747

4848
# bayes_opt
4949

python/seldon/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from .util import RecommenderWrapper,Recommender,Extension,ExtensionWrapper,Recommender_wrapper,Extension_wrapper
44

5-
__version__ = '2.2.5'
5+
__version__ = '2.2.6'
66

77
class NullHandler(logging.Handler):
88
def emit(self, record):

python/seldon/keras.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def default_classification_model(input_width,num_classes):
3434

3535

3636
class 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

Comments
 (0)