from hydroDL import kPath
from hydroDL.app import waterQuality
from hydroDL.master import basins
import numpy as np
dataName = 'sulfateNE'
wqData = waterQuality.DataModelWQ(dataName)
indYrO, indYrE = waterQuality.indYrOddEven(wqData.info)
wqData.saveSubset('Yodd', indYrO)
wqData.saveSubset('Yeven', indYrE)
codeLst = ['00945', '00935']
# subsetLst = ['Yodd', 'Yeven']
subsetLst = ['Yodd']
varXC = ['DRAIN_SQKM', 'SNOW_PCT_PRECIP', 'STREAMS_KM_SQ_KM', 'PCT_1ST_ORDER',
'BFI_AVE', 'CONTACT', 'FORESTNLCD06', 'HLR_BAS_DOM_100M', 'ELEV_MEAN_M_BASIN',
'PERMAVE', 'WTDEPAVE', 'ROCKDEPAVE', 'SLOPE_PCT']
varX1 = ['00060', 'pr', 'sph', 'srad', 'tmmn', 'tmmx',
'pet', 'etr', 'ph', 'Conduc', 'K', 'Cl']
varX2 = ['00060', 'pr', 'sph', 'srad', 'tmmn', 'tmmx',
'pet', 'etr']
nEp = 200
caseLst = list()
for subset in subsetLst:
saveName = '{}-{}-ntn'.format(dataName, subset)
caseName = basins.wrapMaster(
dataName=dataName, trainName=subset, batchSize=[None, 50], nEpoch=nEp,
outName=saveName, varXC=varXC, varX=varX1, varYC=codeLst)
caseLst.append(caseName)
saveName = '{}-{}'.format(dataName, subset)
caseName = basins.wrapMaster(
from hydroDL import kPath
from hydroDL.app import waterQuality
from hydroDL.master import basins
from hydroDL.data import usgs, gageII, gridMET, ntn
import numpy as np
import json
import os
dataName = 'ssW'
wqData = waterQuality.DataModelWQ(dataName)
dirInv = os.path.join(kPath.dirData, 'USGS', 'inventory')
with open(os.path.join(dirInv, 'dictStableSites_0610_0220.json')) as f:
dictSite = json.load(f)
siteNoLst = dictSite['comb']
# create subset
for code in dictSite.keys():
print(code)
siteNoLst = dictSite[code]
b1 = wqData.info['siteNo'].isin(siteNoLst).values
if code != 'comb':
b2 = ~np.isnan(wqData.c[:, wqData.varC.index(code)])
info = wqData.info[b1 & b2]
else:
info = wqData.info[b1]
indY1, indY2 = waterQuality.indYrOddEven(info)
wqData.saveSubset('{}-Y1'.format(code), indY1)
wqData.saveSubset('{}-Y2'.format(code), indY2)