Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
40 commits
Select commit Hold shift + click to select a range
7c8d900
Inital cluster code commit
conformist89 Jul 30, 2021
493f824
Parralel part rehardcoded
conformist89 Jul 30, 2021
dafe099
Transformed code to procedure style
conformist89 Aug 1, 2021
0767fb0
Tranformed everything till comparison with KFPF to procedure style
conformist89 Aug 1, 2021
f29c2a6
Corrected df_dcm_path
conformist89 Aug 3, 2021
d7cebba
1. PNG was added to gitignore 2. preliminary cluster code is implemen…
conformist89 Aug 5, 2021
734bbd4
Corrected some comments to code
conformist89 Aug 5, 2021
5261092
User chooses signal, background and output path by command line argum…
conformist89 Aug 9, 2021
133d9c1
Plotted distributions of trained features
conformist89 Aug 10, 2021
bd19211
Implemented distribution with all the variables
conformist89 Aug 10, 2021
c0f70a9
Corrected definition of B/S after ML cut
conformist89 Aug 11, 2021
2319ad1
Updated Readme.md
conformist89 Aug 11, 2021
54fd113
Corrected Readmi.md
conformist89 Aug 11, 2021
411ed72
Changed legend
conformist89 Aug 11, 2021
5ecd084
Added the same range before and after ML cut
conformist89 Aug 11, 2021
47a597b
Added difference between signal and background
conformist89 Aug 11, 2021
f114f9d
Checked distributions for train dataset
conformist89 Aug 11, 2021
e478420
Trained on 5 features and corrected legend
conformist89 Aug 11, 2021
3bc2251
Works for this set of variables: ['chi2geo', 'chi2primneg', 'chi2prim…
conformist89 Aug 12, 2021
23c0cee
Plotted distributions for deploy data and decreased label's size in c…
conformist89 Aug 13, 2021
9226720
Updated legend
conformist89 Aug 13, 2021
a82eddf
Changes pic's title
conformist89 Aug 14, 2021
06b751e
Implemented function to plot variables before and after ML cut
conformist89 Aug 16, 2021
b9f2846
Plotted distributions for train, test and deploy datasets
conformist89 Aug 16, 2021
27fea11
Implemented function to create dataframe for plotting variables
conformist89 Aug 16, 2021
d33d021
Fixed signal dirrerence(before and after ML cut) computation bug
conformist89 Aug 16, 2021
15d57e8
Added 1D correlation with some variable
conformist89 Aug 16, 2021
bdea651
Implemented pt-rapidity 2d distributions, need to fix difference bug
conformist89 Aug 17, 2021
4317471
Fixed label bug
conformist89 Aug 18, 2021
92f78af
Fixed difference plot bug
conformist89 Aug 18, 2021
955a5cd
Plotted pt-rapidity 2D histogram for train and deploy sets
conformist89 Aug 18, 2021
6cb5145
Implemented preliminary version of 2D hist
conformist89 Aug 19, 2021
b2045a9
Added 2d hist invoke
conformist89 Aug 19, 2021
1d04b24
Deleted comment
conformist89 Aug 19, 2021
e5ca48a
Added name of the sample (for exapmle, train, test, etc)
conformist89 Aug 20, 2021
3504ee5
Added name of the sample (for exapmle, train, test, etc)
conformist89 Aug 20, 2021
3287e0f
Invoked the distributions in the main file
conformist89 Aug 20, 2021
2f9451c
Added samples's name
conformist89 Aug 20, 2021
a88ca35
Preliminary Visualization of the BDT trees architecture
conformist89 Aug 20, 2021
e9c14bf
Invoked 2D pT-rapidity and feature-mass distributions
conformist89 Aug 23, 2021
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
144 changes: 144 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,144 @@
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class

# C extensions
*.so

# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST

# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec

# Installer logs
pip-log.txt
pip-delete-this-directory.txt

# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
*.py,cover
.hypothesis/
.pytest_cache/
cover/

# Translations
*.mo
*.pot

# Django stuff:
*.log
local_settings.py
db.sqlite3
db.sqlite3-journal

# Flask stuff:
instance/
.webassets-cache

# Scrapy stuff:
.scrapy

# Sphinx documentation
docs/_build/

# PyBuilder
.pybuilder/
target/

# Jupyter Notebook
.ipynb_checkpoints

# IPython
profile_default/
ipython_config.py

# pyenv
# For a library or package, you might want to ignore these files since the code is
# intended to run in multiple environments; otherwise, check them in:
# .python-version

# pipenv
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
# However, in case of collaboration, if having platform-specific dependencies or dependencies
# having no cross-platform support, pipenv may install dependencies that don't work, or not
# install all needed dependencies.
#Pipfile.lock

# PEP 582; used by e.g. github.com/David-OConnor/pyflow
__pypackages__/

# Celery stuff
celerybeat-schedule
celerybeat.pid

# SageMath parsed files
*.sage.py

# Environments
.env
.venv
env/
venv/
ENV/
env.bak/
venv.bak/

# Spyder project settings
.spyderproject
.spyproject

# Rope project settings
.ropeproject

# mkdocs documentation
/site

# mypy
.mypy_cache/
.dmypy.json
dmypy.json

# Pyre type checker
.pyre/

# pytype static type analyzer
.pytype/

# Cython debug symbols
cython_debug/

# PNG images
*.png

# PDF files
*.pdf
213 changes: 0 additions & 213 deletions CBM_ML_XGB_package.py

This file was deleted.

5 changes: 4 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,8 +1,11 @@
# XGB
# XGB
This repsoitory contains the basic code of machine learning package XGBoost applied to lambda reconstruction at the CBM experiment at GSI/FAIR.
The code is split into different files so that it is more readible. The library directory contains various files that could be installed locally. For further information visit that directory.

# Tutorial
We have created a Google Colab for our this code and it has detailed information on what has been done in this code and what was the motivation behind it. The colab is available on the following link
https://colab.research.google.com/drive/1Up8YvcWiYv0NN7nOA5YhlpsiqTsY0eCt?usp=sharing

# Usage

python clusterXGB.py /path/to_signal.root /path/to_background.root output_path
Loading