Dataset for Reweighting Wraps a dataset and provides each data-point with a new weight.
More...
#include <DataDriven.h>
|
| ReweightingDataset (const GeneralOptions &general_options, Dataset &dataset, const std::vector< float > &weights) |
| Constructs a new ReweightingDataset.
|
|
virtual unsigned int | getNumberOfFeatures () const override |
| Returns the number of features in this dataset.
|
|
virtual unsigned int | getNumberOfSpectators () const override |
| Returns the number of features in this dataset.
|
|
virtual unsigned int | getNumberOfEvents () const override |
| Returns the number of events in this dataset.
|
|
virtual std::vector< float > | getFeature (unsigned int iFeature) override |
| Returns all values of one feature in a std::vector<float>
|
|
virtual std::vector< float > | getSpectator (unsigned int iSpectator) override |
| Returns all values of one spectator in a std::vector<float>
|
|
virtual std::vector< float > | getWeights () override |
| Returns all weights.
|
|
virtual std::vector< float > | getTargets () override |
| Returns all targets.
|
|
virtual std::vector< bool > | getSignals () override |
| Returns all is Signals.
|
|
virtual void | loadEvent (unsigned int event) override |
| Load the event number iEvent.
|
|
virtual float | getSignalFraction () |
| Returns the signal fraction of the whole sample.
|
|
virtual unsigned int | getFeatureIndex (const std::string &feature) |
| Return index of feature with the given name.
|
|
virtual unsigned int | getSpectatorIndex (const std::string &spectator) |
| Return index of spectator with the given name.
|
|
|
GeneralOptions | m_general_options |
| GeneralOptions passed to this dataset.
|
|
std::vector< float > | m_input |
| Contains all feature values of the currently loaded event.
|
|
std::vector< float > | m_spectators |
| Contains all spectators values of the currently loaded event.
|
|
float | m_weight |
| Contains the weight of the currently loaded event.
|
|
float | m_target |
| Contains the target value of the currently loaded event.
|
|
bool | m_isSignal |
| Defines if the currently loaded event is signal or background.
|
|
Dataset for Reweighting Wraps a dataset and provides each data-point with a new weight.
Definition at line 29 of file DataDriven.h.
◆ ReweightingDataset()
Constructs a new ReweightingDataset.
- Parameters
-
general_options | shared options defining the dataset (variables, ...) |
dataset | containing the data-points |
weights | reweighting weights |
Definition at line 25 of file DataDriven.cc.
26 : Dataset(general_options), m_dataset(dataset), m_weights(weights) { }
◆ getFeature()
virtual std::vector< float > getFeature |
( |
unsigned int | iFeature | ) |
|
|
inlineoverridevirtual |
Returns all values of one feature in a std::vector<float>
- Parameters
-
iFeature | the position of the feature to return |
Reimplemented from Dataset.
Definition at line 59 of file DataDriven.h.
59{ return m_dataset.getFeature(iFeature); }
◆ getFeatureIndex()
unsigned int getFeatureIndex |
( |
const std::string & | feature | ) |
|
|
virtualinherited |
Return index of feature with the given name.
- Parameters
-
feature | name of the feature |
Definition at line 50 of file Dataset.cc.
51 {
52
53 auto it = std::find(m_general_options.m_variables.begin(), m_general_options.m_variables.end(), feature);
54 if (it == m_general_options.m_variables.end()) {
55 B2ERROR("Unknown feature named " << feature);
56 return 0;
57 }
58 return std::distance(m_general_options.m_variables.begin(), it);
59
60 }
◆ getNumberOfEvents()
virtual unsigned int getNumberOfEvents |
( |
| ) |
const |
|
inlineoverridevirtual |
Returns the number of events in this dataset.
Implements Dataset.
Definition at line 53 of file DataDriven.h.
53{ return m_dataset.getNumberOfEvents(); };
◆ getNumberOfFeatures()
virtual unsigned int getNumberOfFeatures |
( |
| ) |
const |
|
inlineoverridevirtual |
Returns the number of features in this dataset.
Implements Dataset.
Definition at line 43 of file DataDriven.h.
43{ return m_dataset.getNumberOfFeatures(); }
◆ getNumberOfSpectators()
virtual unsigned int getNumberOfSpectators |
( |
| ) |
const |
|
inlineoverridevirtual |
Returns the number of features in this dataset.
Implements Dataset.
Definition at line 48 of file DataDriven.h.
48{ return m_dataset.getNumberOfSpectators(); }
◆ getSignalFraction()
float getSignalFraction |
( |
| ) |
|
|
virtualinherited |
Returns the signal fraction of the whole sample.
Reimplemented in SPlotDataset.
Definition at line 35 of file Dataset.cc.
36 {
37
38 double signal_weight_sum = 0;
39 double weight_sum = 0;
40 for (unsigned int i = 0; i < getNumberOfEvents(); ++i) {
41 loadEvent(i);
42 weight_sum += m_weight;
43 if (m_isSignal)
44 signal_weight_sum += m_weight;
45 }
46 return signal_weight_sum / weight_sum;
47
48 }
◆ getSignals()
virtual std::vector< bool > getSignals |
( |
| ) |
|
|
inlineoverridevirtual |
Returns all is Signals.
Reimplemented from Dataset.
Definition at line 87 of file DataDriven.h.
87{ return m_dataset.getSignals(); }
◆ getSpectator()
virtual std::vector< float > getSpectator |
( |
unsigned int | iSpectator | ) |
|
|
inlineoverridevirtual |
Returns all values of one spectator in a std::vector<float>
- Parameters
-
iSpectator | the position of the feature to return |
Reimplemented from Dataset.
Definition at line 65 of file DataDriven.h.
65{ return m_dataset.getSpectator(iSpectator); }
◆ getSpectatorIndex()
unsigned int getSpectatorIndex |
( |
const std::string & | spectator | ) |
|
|
virtualinherited |
Return index of spectator with the given name.
- Parameters
-
spectator | name of the spectator |
Definition at line 62 of file Dataset.cc.
63 {
64
65 auto it = std::find(m_general_options.m_spectators.begin(), m_general_options.m_spectators.end(), spectator);
66 if (it == m_general_options.m_spectators.end()) {
67 B2ERROR("Unknown spectator named " << spectator);
68 return 0;
69 }
70 return std::distance(m_general_options.m_spectators.begin(), it);
71
72 }
◆ getTargets()
virtual std::vector< float > getTargets |
( |
| ) |
|
|
inlineoverridevirtual |
Returns all targets.
Reimplemented from Dataset.
Definition at line 82 of file DataDriven.h.
82{ return m_dataset.getTargets(); }
◆ getWeights()
virtual std::vector< float > getWeights |
( |
| ) |
|
|
inlineoverridevirtual |
Returns all weights.
Reimplemented from Dataset.
Definition at line 70 of file DataDriven.h.
71 {
72 auto weights = m_dataset.getWeights();
73 for (unsigned int iEvent = 0; iEvent < weights.size(); ++iEvent) {
74 weights[iEvent] *= m_weights[iEvent];
75 }
76 return weights;
77 }
◆ loadEvent()
void loadEvent |
( |
unsigned int | event | ) |
|
|
overridevirtual |
Load the event number iEvent.
- Parameters
-
event | event number to load |
Implements Dataset.
Definition at line 28 of file DataDriven.cc.
29 {
30 m_dataset.loadEvent(event);
31 m_input = m_dataset.m_input;
32 m_spectators = m_dataset.m_spectators;
33 m_isSignal = m_dataset.m_isSignal;
34 m_target = m_dataset.m_target;
35 m_weight = m_weights[event] * m_dataset.m_weight;
36 }
◆ m_dataset
◆ m_general_options
◆ m_input
std::vector<float> m_input |
|
inherited |
Contains all feature values of the currently loaded event.
Definition at line 123 of file Dataset.h.
◆ m_isSignal
Defines if the currently loaded event is signal or background.
Definition at line 127 of file Dataset.h.
◆ m_spectators
std::vector<float> m_spectators |
|
inherited |
Contains all spectators values of the currently loaded event.
Definition at line 124 of file Dataset.h.
◆ m_target
Contains the target value of the currently loaded event.
Definition at line 126 of file Dataset.h.
◆ m_weight
Contains the weight of the currently loaded event.
Definition at line 125 of file Dataset.h.
◆ m_weights
std::vector<float> m_weights |
|
private |
The documentation for this class was generated from the following files: