It really depends on your use case.
Could you explain what you are trying to achieve and based on this we could think about the best approach?
E.g. if you derive your custom model form _SimpleSegmentationModel without changing any method, you could just use the parent model directly.