mirror of
https://github.com/QData/TextAttack.git
synced 2021-10-13 00:05:06 +03:00
19 lines
510 B
Python
19 lines
510 B
Python
from argparse import ArgumentParser
|
|
|
|
from textattack.commands import TextAttackCommand
|
|
|
|
|
|
class TrainModelCommand(TextAttackCommand):
|
|
"""
|
|
The TextAttack train module:
|
|
|
|
A command line parser to train a model from user specifications.
|
|
"""
|
|
|
|
def run(self, args):
|
|
raise NotImplementedError("Cannot train models yet - stay tuned!!")
|
|
|
|
@staticmethod
|
|
def register_subcommand(main_parser: ArgumentParser):
|
|
parser = main_parser.add_parser("train", help="train a model")
|