Skip to content

Commit 19ed7b8

Browse files
committed
new set of signals
1 parent 3b5ead4 commit 19ed7b8

File tree

1 file changed

+18
-5
lines changed

1 file changed

+18
-5
lines changed

data/signals.py

Lines changed: 18 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -175,17 +175,30 @@ def fetch_nstx_data(signal_path,shot_num,c):
175175

176176
#for downloading #modify this to preprocess shots with only a subset of signals. This may produce more shots
177177
#since only those shots that contain all_signals contained here are used.
178+
#all_signals = {'q95':q95,'li':li,'ip':ip,
179+
#'betan':betan,'energy':energy,'lm':lm,'dens':dens,
180+
#'pradcore':pradcore,'pradedge':pradedge,'pradtot':pradtot,
181+
#'pin':pin,'torquein':torquein,
182+
#'tmamp1':tmamp1,'tmamp2':tmamp2,'tmfreq1':tmfreq1,'tmfreq2':tmfreq2,
183+
#'pechin':pechin,'energydt':energydt,
184+
#'etemp_profile':etemp_profile,'edens_profile':edens_profile,
185+
# 'ipdirect':ipdirect,'iptarget':iptarget,'iperr':iperr,
186+
# 'etemp_profile_spatial':etemp_profile_spatial,'edens_profile_spatial':edens_profile_spatial,
187+
# 'etemp':etemp
188+
#}
189+
190+
#Restricted subset to those signals that are present for most shots. The idea is to remove signals that cause many shots to be dropped from the dataset.
178191
all_signals = {'q95':q95,'li':li,'ip':ip,
179192
'betan':betan,'energy':energy,'lm':lm,'dens':dens,
180193
'pradcore':pradcore,'pradedge':pradedge,'pradtot':pradtot,
181194
'pin':pin,'torquein':torquein,
182-
'tmamp1':tmamp1,'tmamp2':tmamp2,'tmfreq1':tmfreq1,'tmfreq2':tmfreq2,
183-
'pechin':pechin,'energydt':energydt,
195+
'energydt':energydt,
184196
'etemp_profile':etemp_profile,'edens_profile':edens_profile,
185-
'ipdirect':ipdirect,'iptarget':iptarget,'iperr':iperr,
186-
'etemp_profile_spatial':etemp_profile_spatial,'edens_profile_spatial':edens_profile_spatial,
187-
'etemp':etemp
197+
'ipdirect':ipdirect,'iptarget':iptarget,'iperr':iperr
198+
#'tmamp1':tmamp1,'tmamp2':tmamp2,'tmfreq1':tmfreq1,'tmfreq2':tmfreq2,'pechin':pechin,
199+
# 'etemp_profile_spatial':etemp_profile_spatial,'edens_profile_spatial':edens_profile_spatial,'etemp':etemp
188200
}
201+
189202
#new signals are not downloaded yet
190203

191204

0 commit comments

Comments
 (0)