def test_import_all_models():
""" Try loading all defined models to ensure that their full qualified
names are still good
"""
for model_name in MODELS.keys():
print("Try loading model", model_name)
get_model_class(model_name)
def parse_args(args):
parser = argparse.ArgumentParser()
parser.add_argument(
"--lemmatization",
help="Perform lemmatization (using spaCy)",
action="store_true",
)
parser.add_argument(
"--training-set-size",
nargs="?",
default=14000,
type=int,
help="The size of the training set for the duplicate model",
)
parser.add_argument(
"--disable-url-cleanup",
help="Don't cleanup urls when training the duplicate model",
dest="cleanup_urls",
default=True,
action="store_false",
)
parser.add_argument("--train",
help="Perform training",
action="store_true")
parser.add_argument("--goal",
help="Goal of the classifier",
choices=MODELS.keys(),
default="defect")
parser.add_argument(
"--classifier",
help="Type of the classifier. Only used for component classification.",
choices=["default", "nn"],
default="default",
)
parser.add_argument("--classify",
help="Perform evaluation",
action="store_true")
parser.add_argument(
"--generate-sheet",
help=
"Perform evaluation on bugs from last week and generate a csv file",
action="store_true",
)
parser.add_argument("--token", help="Bugzilla token", action="store")
parser.add_argument(
"--historical",
help="""Analyze historical bugs. Only used for defect, bugtype,
defectenhancementtask and regression tasks.""",
action="store_true",
)
return parser.parse_args(args)
def parse_args(args):
parser = argparse.ArgumentParser()
parser.add_argument(
"--lemmatization",
help="Perform lemmatization (using spaCy)",
action="store_true",
)
parser.add_argument("--train",
help="Perform training",
action="store_true")
parser.add_argument("--goal",
help="Goal of the classifier",
choices=MODELS.keys(),
default="defect")
parser.add_argument(
"--classifier",
help="Type of the classifier. Only used for component classification.",
choices=["default", "nn"],
default="default",
)
parser.add_argument("--classify",
help="Perform evaluation",
action="store_true")
parser.add_argument(
"--generate-sheet",
help=
"Perform evaluation on bugs from last week and generate a csv file",
action="store_true",
)
parser.add_argument("--token", help="Bugzilla token", action="store")
parser.add_argument(
"--historical",
help="""Analyze historical bugs. Only used for defect, bugtype,
defectenhancementtask and regression tasks.""",
action="store_true",
)
return parser.parse_args(args)
from bugbug import bugzilla, db, repository
from bugbug.models import MODELS, get_model_class
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument(
"--lemmatization",
help="Perform lemmatization (using spaCy)",
action="store_true",
)
parser.add_argument("--train",
help="Perform training",
action="store_true")
parser.add_argument("--goal",
help="Goal of the classifier",
choices=MODELS.keys(),
default="defect")
parser.add_argument(
"--classifier",
help="Type of the classifier",
choices=["default", "nn"],
default="default",
)
parser.add_argument("--classify",
help="Perform evaluation",
action="store_true")
parser.add_argument(
"--generate-sheet",
help=
"Perform evaluation on bugs from last week and generate a csv file",
action="store_true",
import numpy as np
from bugbug import repository # noqa
from bugbug import bugzilla, db
from bugbug.models import MODELS, get_model_class
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument(
"--lemmatization",
help="Perform lemmatization (using spaCy)",
action="store_true",
)
parser.add_argument("--train", help="Perform training", action="store_true")
parser.add_argument(
"--goal", help="Goal of the classifier", choices=MODELS.keys(), default="defect"
)
parser.add_argument(
"--classifier",
help="Type of the classifier",
choices=["default", "nn"],
default="default",
)
parser.add_argument("--classify", help="Perform evaluation", action="store_true")
parser.add_argument(
"--generate-sheet",
help="Perform evaluation on bugs from last week and generate a csv file",
action="store_true",
)
parser.add_argument("--token", help="Bugzilla token", action="store")
parser.add_argument(