seldonio / seldon-core-examples Goto Github PK

def predict(self,X,features_names):
        assert X.shape[0]>=1, 'wrong shape 0'
        if X.shape[0]==784:
                  X = X.reshape(1,28,28,1)
        else:
                  X = X.reshape(X.shape[0],28,28,1)

What if I want to make a prediction on a batch of 784 images? As a general guideline, the interface methods can check the format of what they receive but should not assume different formats and implement different behaviours accordingly.

https://github.com/SeldonIO/seldon-core/blob/master/docs/wrappers/readme.md

https://github.com/SeldonIO/seldon-core-examples/blob/master/README.md