You can easily extend MSMBuilder by subclassing BaseEstimator
or any of
its children. You can even build your plugin to work with the msmb
command-line interface.
cmdline.Command
or any of its children. For example,
if you want to expose a new Featurizer from the command line.from msmbuilder.commands.featurizer import FeaturizerCommand
class MyNiftyFeaturizerCommand(FeaturizerCommand):
klass = MyNiftyFeaturizer
_concrete = True
setuptools
.
Use "msmbuilder.commands"
as the entry point.
For example, in your setup.py
.setup(
...
entry_points={'msmbuilder.commands':
'niftyfeat = niftyfeat:MyNiftyFeaturizerCommand'
)
See the setuptools documentation for more information.