@@ -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.
178191all_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