def is_catastrophe(self, obs): X = np.expand_dims(obs, axis=0) X = np.reshape(X, [X.shape[0], -1]) return self.classifier.predict(X)[0]