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(