Platform for Machine Learning projects on Software Engineering
Перейти к файлу
Marco Castelluccio efabd210fd
Fix StructuredColumnTransformer transformer and add some tests for it (#3762)
The test also shows the difference between StructuredColumnTransformer and ColumnTransformer
2023-10-27 17:18:38 +02:00
.github Create a workflow to add new issues to the team project (#3596) 2023-07-26 11:08:28 -04:00
bugbug Fix StructuredColumnTransformer transformer and add some tests for it (#3762) 2023-10-27 17:18:38 +02:00
docs Enable Prettier for JavaScript, HTML, CSS, yaml, Markdown (#2006) 2020-12-17 18:13:49 +01:00
functions/diff2html Bump semver from 5.7.1 to 5.7.2 in /functions/diff2html (#3577) 2023-07-17 11:23:17 -04:00
http_service Bump sentry-sdk[flask] from 1.31.0 to 1.32.0 (#3732) 2023-10-16 12:12:04 +02:00
infra Increase deadline for frontend-build and frontend-deploy tasks in the landings pipeline 2023-10-24 10:36:58 +02:00
scripts Fetch patch from local Mercurial repo and apply to git repo and Stop installing `patch` in our Docker image (#3743) 2023-10-23 17:47:46 -04:00
tests Fix StructuredColumnTransformer transformer and add some tests for it (#3762) 2023-10-27 17:18:38 +02:00
ui/changes Bump apexcharts from 3.43.0 to 3.44.0 in /ui/changes (#3744) 2023-10-23 20:13:36 +02:00
.codecov.yml Disable codecov pull request comments 2021-01-11 17:16:11 +01:00
.dockerignore Ignore http_service directory when building bugbug base images (#1338) 2020-02-28 19:21:28 +01:00
.flake8 Use flake8 configuration suggested by black 2023-04-17 12:12:56 +02:00
.gitignore Import changes UI (#1940) 2020-11-21 00:24:02 +01:00
.isort.cfg Use profile=black for isort 2023-04-17 12:06:05 +02:00
.pre-commit-config.yaml Update some pre-commit repositories 2023-10-11 10:08:09 +02:00
.prettierrc Use the same Prettier configuration as mozilla-central 2021-03-10 09:18:07 +01:00
.taskcluster.yml Switch Taskcluster to use GitHub checks API (#3719) 2023-10-14 00:57:30 +02:00
CITATION.cff Update pre-commit repositories 2023-04-17 11:18:04 +02:00
CODE_OF_CONDUCT.md Enable Prettier for JavaScript, HTML, CSS, yaml, Markdown (#2006) 2020-12-17 18:13:49 +01:00
CONTRIBUTING.md Add a link to Discussions in CONTRIBUTING.md (#3333) 2023-03-14 07:41:16 -04:00
LICENSE First commit 2018-03-11 20:12:35 +00:00
MANIFEST.in Move Keras as an optional dependency (#319) 2019-04-26 18:50:02 +02:00
README.md Update the readme file to require python 3.10+ (#3706) 2023-10-11 10:10:06 +02:00
VERSION Version 0.0.522 2023-10-19 00:14:04 +02:00
docker-compose.yml Add support for GitHub issue classification to the HTTP service (#2330) 2021-05-11 08:48:07 +02:00
extra-nlp-requirements.txt Bump spacy from 3.7.1 to 3.7.2 (#3738) 2023-10-18 23:43:23 +02:00
extra-nn-requirements.txt Bump tensorflow from 2.13.0 to 2.14.0 (#3680) 2023-10-03 10:24:26 -04:00
requirements.txt Revert "Bump pandas from 2.1.0 to 2.1.1" (#3753) 2023-10-24 09:53:59 +02:00
setup.py Run tests on Python 3.10 (#3571) 2023-07-07 01:10:57 +02:00
test-requirements.txt Bump hypothesis from 6.88.0 to 6.88.1 (#3739) 2023-10-18 23:43:16 +02:00

README.md

bugbug

Task Status codecov

Bugbug aims at leveraging machine learning techniques to help with bug and quality management, and other software engineering tasks (such as test selection and defect prediction).

Chat with us in the bugbug Matrix room.

More information on the Mozilla Hacks blog:

Classifiers

  • assignee - The aim of this classifier is to suggest an appropriate assignee for a bug.

  • backout - The aim of this classifier is to detect patches that might be more likely to be backed-out (because of build or test failures). It could be used for test prioritization/scheduling purposes.

  • bugtype - The aim of this classifier is to classify bugs according to their type. The labels are gathered automatically from bugs: right now they are "crash/memory/performance/security". The plan is to add more types after manual labeling.

  • component - The aim of this classifier is to assign product/component to (untriaged) bugs.

  • defect vs enhancement vs task - Extension of the defect classifier to detect differences also between feature requests and development tasks.

  • defect - Bugs on Bugzilla aren't always bugs. Sometimes they are feature requests, refactorings, and so on. The aim of this classifier is to distinguish between bugs that are actually bugs and bugs that aren't. The dataset currently contains 2110 bugs, the accuracy of the current classifier is ~93% (precision ~95%, recall ~94%).

  • devdocneeded - The aim of this classifier is to detect bugs which should be documented for developers.

  • duplicate - The aim of this classifier is to detect duplicate bugs.

  • needsdiagnosis - The aim of this classifier is to detect issues that are likely invalid and don't need to be diagnosed for webcompat use case.

  • qaneeded - The aim of this classifier is to detect bugs that would need QA verification.

  • regression vs non-regression - Bugzilla has a regression keyword to identify bugs that are regressions. Unfortunately it isn't used consistently. The aim of this classifier is to detect bugs that are regressions.

  • regressionrange - The aim of this classifier is to detect regression bugs that have a regression range vs those that don't.

  • regressor - The aim of this classifier is to detect patches which are more likely to cause regressions. It could be used to make riskier patches undergo more scrutiny.

  • spam - The aim of this classifier is to detect bugs which are spam.

  • stepstoreproduce - The aim of this classifier is to detect bugs that have steps to reproduce vs those that don't.

  • testfailure - The aim of this classifier is to detect patches that might be more likely to cause test failures.

  • testselect - The aim of this classifier is to select relevant tests to run for a given patch.

  • tracking - The aim of this classifier is to detect bugs to track.

  • uplift - The aim of this classifier is to detect bugs for which uplift should be approved and bugs for which uplift should not be approved.

Setup and Prerequisites

Install the Python dependencies:

pip3 install -r requirements.txt

You may also need pip install -r test-requirements.txt. Depending on the parts of bugbug you want to run, you might need to install dependencies from other requirement files (find them with find . -name "*requirements*").

Currently, Python 3.10+ is required. You can double check the version we use by looking at setup.py.

Also, libgit2 (needs v1.0.0, only in experimental on Debian), might be required (if you can't install it, skip this step).

sudo apt-get -t experimental install libgit2-dev

Auto-formatting

This project is using pre-commit. Please run pre-commit install to install the git pre-commit hooks on your clone.

Every time you will try to commit, pre-commit will run checks on your files to make sure they follow our style standards and they aren't affected by some simple issues. If the checks fail, pre-commit won't let you commit.

Usage

Training

Run the trainer.py script with the command python -m scripts.trainer (with --help to see the required and optional arguments of the command) to perform training (warning this takes 30min+).

Testing

To use a model to classify a given bug, you can run python -m scripts.bug_classifier MODEL_NAME --bug-id ID_OF_A_BUG_FROM_BUGZILLA. N.B.: If you run the classifier script without training a model first, it will automatically download an already trained model.

Example for the "defect" model

training To train the model for mode defect:

python3 -m scripts.trainer defect

testing To use the model to classify a given bug, you can run python -m scripts.bug_classifier defect --bug-id ID_OF_A_BUG_FROM_BUGZILLA.

Running the repository mining script

Note: This section is only necessary if you want to perform changes to the repository mining script. Otherwise, you can simply use the commits data we generate automatically.

  1. Clone https://hg.mozilla.org/mozilla-central/.
  2. Run ./mach vcs-setup in the directory where you have cloned mozilla-central.
  3. Enable the extensions mentioned in infra/hgrc. For example, if you are on Linux, you can add firefoxtree to the extensions section of the ~/.hgrc file as:
    firefoxtree = ~/.mozbuild/version-control-tools/hgext/firefoxtree
    
  4. Run the repository.py script, with the only argument being the path to the mozilla-central repository.

Note: If you run into problems, it's possible the version of Mercurial you are using is not supported. Check the Docker definition at infra/dockerfile.commit_retrieval to see what we are using in production.

Note: the script will take a long time to run (on my laptop more than 7 hours). If you want to test a simple change and you don't intend to actually mine the data, you can modify the repository.py script to limit the number of analyzed commits. Simply add limit=1024 to the call to the log command.

Structure of the project

  • bugbug/labels contains manually collected labels;
  • bugbug/db.py is an implementation of a really simple JSON database;
  • bugbug/bugzilla.py contains the functions to retrieve bugs from the Bugzilla tracking system;
  • bugbug/repository.py contains the functions to mine data from the mozilla-central (Firefox) repository;
  • bugbug/bug_features.py contains functions to extract features from bug/commit data;
  • bugbug/model.py contains the base class that all models derive from;
  • bugbug/models contains implementations of specific models;
  • bugbug/nn.py contains utility functions to include Keras models into a scikit-learn pipeline;
  • bugbug/utils.py contains misc utility functions;
  • bugbug/nlp contains utility functions for NLP;
  • bugbug/labels.py contains utility functions for handling labels;
  • bugbug/bug_snapshot.py contains a module to play back the history of a bug;
  • bugbug/github.py contains functions to retrieve issues from GitHub for a specified owner/repository.

Using bugbug for non-Mozilla projects

Bugbug is focussing on Mozilla use-cases for Firefox, Bugzilla and GitHub. However, we will be happy to accept pull requests adding support for other projects or bug trackers.