def main(args):
# Initialise
args, cfg = initialise(args)
# Load data
data, features, _ = load_data(args.input + 'data.h5',
train=True,
background=True)
# Fill Tau21 profile
profile = fill_profile(data, VAR_TAU21)
# Fit profile
fit = ROOT.TF1('fit', 'pol1', *FIT_RANGE)
profile.Fit('fit', 'RQ0')
intercept_val, coef_val = fit.GetParameter(0), fit.GetParameter(1)
intercept_err, coef_err = fit.GetParError(0), fit.GetParError(1)
# Create scikit-learn transform
ddt = LinearRegression()
ddt.coef_ = np.array([coef_val])
ddt.intercept_ = np.array([-coef_val * FIT_RANGE[0]])
ddt.offset_ = np.array([coef_val * FIT_RANGE[0] + intercept_val])
print "Fitted function:"
print " intercept: {:7.4f} ± {:7.4f}".format(intercept_val, intercept_err)
print " coef: {:7.4f} ± {:7.4f}".format(coef_val, coef_err)
# Save DDT transform
saveclf(ddt, 'models/ddt/ddt.pkl.gz')
return 0
def main(args):
# Initialise
args, cfg = initialise(args)
# Load data
data, features, _ = load_data(args.input + 'data.h5',
train=True,
background=True)
#variable = VAR_TAU21
variable = VAR_N2
#variable = VAR_DECDEEP
#variable = VAR_DEEP
# Fill variable profile
profile = fill_profile(data, variable)
# Fit profile
if variable == VAR_N2:
fit_range = FIT_RANGE_N2
elif variable == VAR_TAU21:
fit_range = FIT_RANGE_TAU21
elif variable == VAR_DECDEEP:
fit_range = FIT_RANGE_DECDEEP
elif variable == VAR_DEEP:
fit_range = FIT_RANGE_DEEP
else:
print "variable invalid"
return 0
fit = ROOT.TF1('fit', 'pol1', *fit_range)
profile.Fit('fit', 'RQ0')
intercept_val, coef_val = fit.GetParameter(0), fit.GetParameter(1)
intercept_err, coef_err = fit.GetParError(0), fit.GetParError(1)
# Create scikit-learn transform
ddt = LinearRegression()
ddt.coef_ = np.array([coef_val])
ddt.intercept_ = np.array([-coef_val * fit_range[0]])
ddt.offset_ = np.array([coef_val * fit_range[0] + intercept_val])
print "Fitted function:"
print " intercept: {:7.4f} ± {:7.4f}".format(intercept_val, intercept_err)
print " coef: {:7.4f} ± {:7.4f}".format(coef_val, coef_err)
# Save DDT transform
saveclf(ddt, 'models/ddt/ddt_{}.pkl.gz'.format(variable))
print "got to the end of main()"
return 0
