Skip to content

Commit

Permalink
Fixed redshift_binned_spectrum to actually redshift specturm.
Browse files Browse the repository at this point in the history
Changed training_parameters.py to 3000A (instead of 3500) as this was what was used in model_v04 (despite model_info saying otherwise).
  • Loading branch information
daniel-muthukrishna committed Jul 30, 2018
1 parent c325b2a commit e0c0a61
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 7 deletions.
13 changes: 7 additions & 6 deletions dash/helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,8 @@ def calc_params_for_log_redshifting(dataDirName):
with open(os.path.join(dataDirName, 'training_params.pickle'), 'rb') as f1:
pars = pickle.load(f1)
w0, w1, nw = pars['w0'], pars['w1'], pars['nw']

n = np.arange(0, int(nw))

dwlog = np.log(w1 / w0) / nw
# wlog = w0 * np.exp(n * dwlog)
# waveInterp = scipy.interpolate.interp1d(n, wlog) # Will interpolate any value within wavelength given an index
Expand All @@ -22,12 +22,13 @@ def redshift_binned_spectrum(flux, z, nIndexes, dwlog, w0, w1, nw, outerVal=0.5)
# assert len(flux) == nw
redshiftedIndexes = nIndexes + np.log(1 + z) / dwlog
indexesInRange = redshiftedIndexes[redshiftedIndexes > 0]
fluxInterp = scipy.interpolate.interp1d(indexesInRange, flux[redshiftedIndexes>0], kind='linear')
fluxInterp = scipy.interpolate.interp1d(indexesInRange, flux[redshiftedIndexes > 0], kind='linear')

minWaveIndex = int(indexesInRange[0])

maxWaveIndex = len(indexesInRange)
fluxRedshifted = np.zeros(nw)
fluxRedshifted[0:maxWaveIndex] = fluxInterp(indexesInRange)
fluxRedshifted[maxWaveIndex:] = outerVal * np.ones(nw-maxWaveIndex)
fluxRedshifted[0:minWaveIndex] = outerVal * np.ones(minWaveIndex)
fluxRedshifted[minWaveIndex:] = fluxInterp(indexesInRange)[:nw-minWaveIndex]

# Apodize edges
preprocess = PreProcessSpectrum(w0, w1, nw)
Expand All @@ -54,4 +55,4 @@ def div0(a, b):
with np.errstate(divide='ignore', invalid='ignore'):
c = np.true_divide(a, b)
c[~ np.isfinite(c)] = 0 # -inf inf NaN
return c
return c
2 changes: 1 addition & 1 deletion dash/training_parameters.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ def create_training_params_file(dataDirName):
'Ib-norm', 'Ibn', 'IIb', 'Ib-pec', 'Ic-norm', 'Ic-broad',
'Ic-pec', 'IIP', 'IIL', 'IIn', 'II-pec'],
'nTypes': 17,
'w0': 3500., # wavelength range in Angstroms
'w0': 3000., # wavelength range in Angstroms
'w1': 10000.,
'nw': 1024, # number of wavelength bins
'minAge': -20.,
Expand Down

0 comments on commit e0c0a61

Please sign in to comment.