model_wrapper

class mastml.models.model_wrapper(model)[source]

Bases: object

Wrapper for pytorch model to include predict method

Methods Summary

Methods Documentation

predict(X)[source]