17427 11 months ago
parent 8e74682902
commit 1e01fb30ff

@ -1,166 +0,0 @@
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': -0.00039}
{'city': '丽水', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00186}
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': -0.00064}
{'city': '丽水', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00213}
{'city': '丽水', 'industry': '第一产业', 'month_deviation_rate': 0.00021}
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 7e-05}
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 0.02438}
{'city': '丽水', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00707}
{'city': '丽水', 'industry': '第一产业', 'month_deviation_rate': -0.00288}
{'city': '丽水', 'industry': '5.农、林、牧、渔专业及辅助性活动', 'month_deviation_rate': -0.0136}
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 0.00028}
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': -0.00094}
{'city': '丽水', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00156}
{'city': '丽水', 'industry': '第一产业', 'month_deviation_rate': 0.00288}
{'city': '丽水', 'industry': '5.农、林、牧、渔专业及辅助性活动', 'month_deviation_rate': 0.00713}
{'city': '丽水', 'industry': '3.酒、饮料及精制茶制造业', 'month_deviation_rate': 0.0084}
{'city': '丽水', 'industry': '10.造纸和纸制品业', 'month_deviation_rate': 0.00644}
{'city': '丽水', 'industry': '15.医药制造业', 'month_deviation_rate': 0.00904}
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 0.00011}
{'city': '丽水', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00193}
{'city': '丽水', 'industry': '第一产业', 'month_deviation_rate': -8e-05}
{'city': '丽水', 'industry': '5.农、林、牧、渔专业及辅助性活动', 'month_deviation_rate': 0.01228}
{'city': '丽水', 'industry': '3.酒、饮料及精制茶制造业', 'month_deviation_rate': 0.00667}
{'city': '丽水', 'industry': '10.造纸和纸制品业', 'month_deviation_rate': 0.0072}
{'city': '丽水', 'industry': '15.医药制造业', 'month_deviation_rate': 0.00769}
{'city': '丽水', 'industry': '20.有色金属冶炼和压延加工业', 'month_deviation_rate': -0.00297}
{'city': '丽水', 'industry': '27.计算机、通信和其他电子设备制造业', 'month_deviation_rate': -0.00224}
{'city': '丽水', 'industry': '2.燃气生产和供应业', 'month_deviation_rate': -0.00205}
{'city': '丽水', 'industry': '四、交通运输、仓储和邮政业', 'month_deviation_rate': 0.00111}
{'city': '丽水', 'industry': '5.管道运输业', 'month_deviation_rate': 0.00114}
{'city': '丽水', 'industry': '2.互联网和相关服务', 'month_deviation_rate': 0.00095}
{'city': '丽水', 'industry': '十一、公共服务及管理组织', 'month_deviation_rate': 0.0029}
{'city': '丽水', 'industry': 'a、全行业用电合计', 'month_deviation_rate': 0.00366}
{'city': '丽水', 'industry': '第二产业', 'month_deviation_rate': 0.00292}
{'city': '丽水', 'industry': '第三产业', 'month_deviation_rate': -0.00145}
{'city': '丽水', 'industry': '城镇居民', 'month_deviation_rate': -0.00088}
{'city': '丽水', 'industry': '一、农、林、牧、渔业', 'month_deviation_rate': 0.00278}
{'city': '丽水', 'industry': '1.农业', 'month_deviation_rate': 0.00376}
{'city': '丽水', 'industry': '3.畜牧业', 'month_deviation_rate': 0.00099}
{'city': '丽水', 'industry': '二、工业', 'month_deviation_rate': 0.002}
{'city': '丽水', 'industry': '(一)采矿业', 'month_deviation_rate': -0.00226}
{'city': '丽水', 'industry': '2.石油和天然气开采业', 'month_deviation_rate': 0.00666}
{'city': '丽水', 'industry': '3.黑色金属矿采选业', 'month_deviation_rate': -0.00379}
{'city': '丽水', 'industry': '5.非金属矿采选业', 'month_deviation_rate': -0.00887}
{'city': '丽水', 'industry': '(二)制造业', 'month_deviation_rate': 0.00446}
{'city': '丽水', 'industry': '1.农副食品加工业', 'month_deviation_rate': 0.00026}
{'city': '丽水', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 0.00127}
{'city': '丽水', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00184}
{'city': '丽水', 'industry': '第一产业', 'month_deviation_rate': 0.00155}
{'city': '丽水', 'industry': '5.农、林、牧、渔专业及辅助性活动', 'month_deviation_rate': 0.01164}
{'city': '丽水', 'industry': '3.酒、饮料及精制茶制造业', 'month_deviation_rate': 0.0067}
{'city': '丽水', 'industry': '10.造纸和纸制品业', 'month_deviation_rate': 0.00705}
{'city': '丽水', 'industry': '15.医药制造业', 'month_deviation_rate': 0.00027}
{'city': '丽水', 'industry': '20.有色金属冶炼和压延加工业', 'month_deviation_rate': 0.00203}
{'city': '丽水', 'industry': '27.计算机、通信和其他电子设备制造业', 'month_deviation_rate': -0.01}
{'city': '丽水', 'industry': '2.燃气生产和供应业', 'month_deviation_rate': 0.00526}
{'city': '丽水', 'industry': '四、交通运输、仓储和邮政业', 'month_deviation_rate': 0.00254}
{'city': '丽水', 'industry': '5.管道运输业', 'month_deviation_rate': -0.02284}
{'city': '丽水', 'industry': '2.互联网和相关服务', 'month_deviation_rate': 0.00142}
{'city': '丽水', 'industry': '十一、公共服务及管理组织', 'month_deviation_rate': -0.00062}
{'city': '丽水', 'industry': 'a、全行业用电合计', 'month_deviation_rate': 0.00339}
{'city': '丽水', 'industry': '第二产业', 'month_deviation_rate': 0.00429}
{'city': '丽水', 'industry': '第三产业', 'month_deviation_rate': -0.00028}
{'city': '丽水', 'industry': '城镇居民', 'month_deviation_rate': 0.00026}
{'city': '丽水', 'industry': '一、农、林、牧、渔业', 'month_deviation_rate': 0.00229}
{'city': '丽水', 'industry': '1.农业', 'month_deviation_rate': 0.00335}
{'city': '丽水', 'industry': '3.畜牧业', 'month_deviation_rate': -0.00269}
{'city': '丽水', 'industry': '二、工业', 'month_deviation_rate': 0.00222}
{'city': '丽水', 'industry': '(一)采矿业', 'month_deviation_rate': -0.00122}
{'city': '丽水', 'industry': '2.石油和天然气开采业', 'month_deviation_rate': 0.00658}
{'city': '丽水', 'industry': '3.黑色金属矿采选业', 'month_deviation_rate': -0.00385}
{'city': '丽水', 'industry': '5.非金属矿采选业', 'month_deviation_rate': -0.00753}
{'city': '丽水', 'industry': '(二)制造业', 'month_deviation_rate': 0.00311}
{'city': '丽水', 'industry': '1.农副食品加工业', 'month_deviation_rate': 0.00098}
{'city': '丽水', 'industry': '5.纺织业', 'month_deviation_rate': -0.00254}
{'city': '丽水', 'industry': '6.纺织服装、服饰业', 'month_deviation_rate': 0.00886}
{'city': '丽水', 'industry': '7.皮革、毛皮、羽毛及其制品和制鞋业', 'month_deviation_rate': -0.00594}
{'city': '丽水', 'industry': '8.木材加工和木、竹、藤、棕、草制品业', 'month_deviation_rate': 0.00454}
{'city': '丽水', 'industry': '9.家具制造业', 'month_deviation_rate': 0.00351}
{'city': '丽水', 'industry': '11.印刷和记录媒介复制业', 'month_deviation_rate': 0.00117}
{'city': '丽水', 'industry': '12.文教、工美、体育和娱乐用品制造业', 'month_deviation_rate': 0.01259}
{'city': '丽水', 'industry': '其中:体育用品制造', 'month_deviation_rate': 0.00271}
{'city': '丽水', 'industry': '13.石油、煤炭及其他燃料加工业', 'month_deviation_rate': -0.00457}
{'city': '丽水', 'industry': '其中:煤化工', 'month_deviation_rate': -8e-05}
{'city': '丽水', 'industry': '14.化学原料和化学制品制造业', 'month_deviation_rate': -0.0086}
{'city': '丽水', 'industry': '其中:肥料制造', 'month_deviation_rate': 0.00571}
{'city': '丽水', 'industry': '其中:中成药生产', 'month_deviation_rate': -0.00341}
{'city': '丽水', 'industry': '16.化学纤维制造业', 'month_deviation_rate': -0.02107}
{'city': '丽水', 'industry': '17.橡胶和塑料制品业', 'month_deviation_rate': 0.00064}
{'city': '丽水', 'industry': '其中:橡胶制品业', 'month_deviation_rate': 0.00124}
{'city': '丽水', 'industry': '塑料制品业', 'month_deviation_rate': -0.00086}
{'city': '丽水', 'industry': '18.非金属矿物制品业', 'month_deviation_rate': -0.00679}
{'city': '丽水', 'industry': '其中:水泥制造', 'month_deviation_rate': -0.00816}
{'city': '丽水', 'industry': '19.黑色金属冶炼和压延加工业', 'month_deviation_rate': 0.00146}
{'city': '丽水', 'industry': '其中:钢铁', 'month_deviation_rate': -0.00054}
{'city': '丽水', 'industry': '铁合金冶炼', 'month_deviation_rate': -0.01302}
{'city': '丽水', 'industry': '21.金属制品业', 'month_deviation_rate': 0.00288}
{'city': '丽水', 'industry': '其中:结构性金属制品制造', 'month_deviation_rate': 0.00654}
{'city': '丽水', 'industry': '22.通用设备制造业', 'month_deviation_rate': -0.00295}
{'city': '丽水', 'industry': '23.专用设备制造业', 'month_deviation_rate': -0.00766}
{'city': '丽水', 'industry': '其中:医疗仪器设备及器械制造', 'month_deviation_rate': -0.00552}
{'city': '丽水', 'industry': '24.汽车制造业', 'month_deviation_rate': 0.00274}
{'city': '丽水', 'industry': '25.铁路、船舶、航空航天和其他运输设备制造业', 'month_deviation_rate': -0.00186}
{'city': '丽水', 'industry': '其中:铁路运输设备制造', 'month_deviation_rate': -0.01057}
{'city': '丽水', 'industry': '26.电气机械和器材制造业', 'month_deviation_rate': -0.00101}
{'city': '丽水', 'industry': '其中:光伏设备及元器件制造', 'month_deviation_rate': 0.00725}
{'city': '丽水', 'industry': '其中:计算机制造', 'month_deviation_rate': 0.01069}
{'city': '丽水', 'industry': '通信设备制造', 'month_deviation_rate': 0.04492}
{'city': '丽水', 'industry': '28.仪器仪表制造业', 'month_deviation_rate': -0.0103}
{'city': '丽水', 'industry': '29.其他制造业', 'month_deviation_rate': 0.00434}
{'city': '丽水', 'industry': '30.废弃资源综合利用业', 'month_deviation_rate': -0.00562}
{'city': '丽水', 'industry': '31.金属制品、机械和设备修理业', 'month_deviation_rate': 0.00644}
{'city': '丽水', 'industry': '(三)电力、热力、燃气及水的生产和供应业', 'month_deviation_rate': 0.01762}
{'city': '丽水', 'industry': '1.电力、热力生产和供应业', 'month_deviation_rate': 0.01065}
{'city': '丽水', 'industry': '其中:电厂生产全部耗用电量', 'month_deviation_rate': 0.05459}
{'city': '丽水', 'industry': '线路损失电量', 'month_deviation_rate': -0.00855}
{'city': '丽水', 'industry': '抽水蓄能抽水耗用电量', 'month_deviation_rate': -0.02152}
{'city': '丽水', 'industry': '3.水的生产和供应业', 'month_deviation_rate': -0.0111}
{'city': '丽水', 'industry': '三、建筑业', 'month_deviation_rate': -0.00484}
{'city': '丽水', 'industry': '1.房屋建筑业', 'month_deviation_rate': -0.00304}
{'city': '丽水', 'industry': '2.土木工程建筑业', 'month_deviation_rate': -0.00242}
{'city': '丽水', 'industry': '3.建筑安装业', 'month_deviation_rate': -0.00415}
{'city': '丽水', 'industry': '4.建筑装饰、装修和其他建筑业', 'month_deviation_rate': -0.00443}
{'city': '丽水', 'industry': '1.铁路运输业', 'month_deviation_rate': 0.00325}
{'city': '丽水', 'industry': '2.道路运输业', 'month_deviation_rate': 0.00499}
{'city': '丽水', 'industry': '其中:城市公共交通运输', 'month_deviation_rate': 0.00494}
{'city': '丽水', 'industry': '3.水上运输业', 'month_deviation_rate': 0.01894}
{'city': '丽水', 'industry': '4.航空运输业', 'month_deviation_rate': 0.00022}
{'city': '丽水', 'industry': '6.多式联运和运输代理业', 'month_deviation_rate': 0.01112}
{'city': '丽水', 'industry': '7.装卸搬运和仓储业', 'month_deviation_rate': -0.00373}
{'city': '丽水', 'industry': '8.邮政业', 'month_deviation_rate': -0.00415}
{'city': '丽水', 'industry': '五、信息传输、软件和信息技术服务业', 'month_deviation_rate': -0.00055}
{'city': '丽水', 'industry': '1.电信、广播电视和卫星传输服务', 'month_deviation_rate': -0.00103}
{'city': '丽水', 'industry': '其中:互联网数据服务', 'month_deviation_rate': -0.00549}
{'city': '丽水', 'industry': '3.软件和信息技术服务业', 'month_deviation_rate': 0.00119}
{'city': '丽水', 'industry': '六、批发和零售业', 'month_deviation_rate': 0.00068}
{'city': '丽水', 'industry': '其中:充换电服务业', 'month_deviation_rate': -0.00012}
{'city': '丽水', 'industry': '七.住宿和餐饮业', 'month_deviation_rate': -0.00306}
{'city': '丽水', 'industry': '八、金融业', 'month_deviation_rate': -0.00339}
{'city': '丽水', 'industry': '十、租赁和商务服务业', 'month_deviation_rate': -0.00144}
{'city': '丽水', 'industry': '其中:租赁业', 'month_deviation_rate': 0.00491}
{'city': '丽水', 'industry': '1.科学研究和技术服务业', 'month_deviation_rate': 0.00129}
{'city': '丽水', 'industry': '其中:科技推广和应用服务业', 'month_deviation_rate': 0.00556}
{'city': '丽水', 'industry': '2.水利、环境和公共设施管理业', 'month_deviation_rate': -0.00364}
{'city': '丽水', 'industry': '其中:公共照明', 'month_deviation_rate': -0.00108}
{'city': '丽水', 'industry': '3.居民服务、修理和其他服务业', 'month_deviation_rate': -0.00052}
{'city': '丽水', 'industry': '其中:教育', 'month_deviation_rate': -0.0295}
{'city': '丽水', 'industry': '5.卫生和社会工作', 'month_deviation_rate': 0.00139}
{'city': '丽水', 'industry': '4.教育、文化、体育和娱乐业', 'month_deviation_rate': -0.00204}
{'city': '丽水', 'industry': '6.公共管理和社会组织、国际组织', 'month_deviation_rate': -0.00557}
{'city': '丽水', 'industry': '乡村居民', 'month_deviation_rate': 0.00099}
{'city': '丽水', 'industry': '2.林业', 'month_deviation_rate': 0.00595}
{'city': '丽水', 'industry': '4.渔业', 'month_deviation_rate': 0.01508}
{'city': '丽水', 'industry': '其中:排灌', 'month_deviation_rate': 0.0038}
{'city': '丽水', 'industry': '1.煤炭开采和洗选业', 'month_deviation_rate': 0.03555}
{'city': '丽水', 'industry': '4.有色金属矿采选业', 'month_deviation_rate': -0.00336}
{'city': '丽水', 'industry': '6.其他采矿业', 'month_deviation_rate': 0.00659}
{'city': '丽水', 'industry': '2.食品制造业', 'month_deviation_rate': 0.00486}
{'city': '丽水', 'industry': '4.烟草制品业', 'month_deviation_rate': 0.15657}
{'city': '丽水', 'industry': '九、房地产业', 'month_deviation_rate': -0.00152}
{'city': '丽水', 'industry': '其中:水利管理业', 'month_deviation_rate': -0.00044}
{'city': '丽水', 'industry': '生物药品制品制造', 'month_deviation_rate': -0.00296}
{'city': '丽水', 'industry': '玻璃制造', 'month_deviation_rate': -0.01427}
{'city': '丽水', 'industry': '陶瓷制品制造', 'month_deviation_rate': 0.00576}
{'city': '丽水', 'industry': '其中:电气化铁路', 'month_deviation_rate': 0.00064}

@ -1,61 +0,0 @@
{'city': '台州', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 0.00089}
{'city': '台州', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00088}
{'city': '台州', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 3e-05}
{'city': '台州', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00038}
{'city': '台州', 'industry': '第一产业', 'month_deviation_rate': -0.00188}
{'city': '台州', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 0.0233}
{'city': '台州', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00211}
{'city': '台州', 'industry': '第一产业', 'month_deviation_rate': -0.00884}
{'city': '台州', 'industry': '5.农、林、牧、渔专业及辅助性活动', 'month_deviation_rate': -0.00474}
{'city': '台州', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': -0.00016}
{'city': '台州', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': 0.00035}
{'city': '台州', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00019}
{'city': '台州', 'industry': '第一产业', 'month_deviation_rate': -0.00105}
{'city': '台州', 'industry': '5.农、林、牧、渔专业及辅助性活动', 'month_deviation_rate': -0.00152}
{'city': '台州', 'industry': '3.酒、饮料及精制茶制造业', 'month_deviation_rate': 0.01137}
{'city': '台州', 'industry': '10.造纸和纸制品业', 'month_deviation_rate': 0.00135}
{'city': '台州', 'industry': '15.医药制造业', 'month_deviation_rate': -0.00079}
{'city': '台州', 'industry': 'b、城乡居民生活用电合计', 'month_deviation_rate': -0.00365}
{'city': '台州', 'industry': '全社会用电总计', 'month_deviation_rate': 0.00055}
{'city': '台州', 'industry': '第一产业', 'month_deviation_rate': -0.00487}
{'city': '台州', 'industry': '5.农、林、牧、渔专业及辅助性活动', 'month_deviation_rate': -0.00023}
{'city': '台州', 'industry': '3.酒、饮料及精制茶制造业', 'month_deviation_rate': 0.01423}
{'city': '台州', 'industry': '10.造纸和纸制品业', 'month_deviation_rate': 0.00156}
{'city': '台州', 'industry': '15.医药制造业', 'month_deviation_rate': -0.00088}
{'city': '台州', 'industry': '20.有色金属冶炼和压延加工业', 'month_deviation_rate': -0.0005}
{'city': '台州', 'industry': '27.计算机、通信和其他电子设备制造业', 'month_deviation_rate': 0.00273}
{'city': '台州', 'industry': '2.燃气生产和供应业', 'month_deviation_rate': -0.00151}
{'city': '台州', 'industry': '四、交通运输、仓储和邮政业', 'month_deviation_rate': 0.00332}
{'city': '台州', 'industry': '5.管道运输业', 'month_deviation_rate': -0.0257}
{'city': '台州', 'industry': '2.互联网和相关服务', 'month_deviation_rate': 0.0002}
{'city': '台州', 'industry': '十一、公共服务及管理组织', 'month_deviation_rate': 0.00158}
{'city': '台州', 'industry': 'a、全行业用电合计', 'month_deviation_rate': 0.00093}
{'city': '台州', 'industry': '第二产业', 'month_deviation_rate': 0.00091}
{'city': '台州', 'industry': '第三产业', 'month_deviation_rate': 0.00155}
{'city': '台州', 'industry': '城镇居民', 'month_deviation_rate': 0.00185}
{'city': '台州', 'industry': '一、农、林、牧、渔业', 'month_deviation_rate': 0.00086}
{'city': '台州', 'industry': '1.农业', 'month_deviation_rate': -0.00079}
{'city': '台州', 'industry': '3.畜牧业', 'month_deviation_rate': -0.00535}
{'city': '台州', 'industry': '二、工业', 'month_deviation_rate': 0.00227}
{'city': '台州', 'industry': '(一)采矿业', 'month_deviation_rate': 0.00376}
{'city': '台州', 'industry': '2.石油和天然气开采业', 'month_deviation_rate': 9e-05}
{'city': '台州', 'industry': '5.非金属矿采选业', 'month_deviation_rate': 0.00012}
{'city': '台州', 'industry': '(二)制造业', 'month_deviation_rate': -8e-05}
{'city': '台州', 'industry': '1.农副食品加工业', 'month_deviation_rate': 0.00024}
{'city': '台州', 'industry': '5.纺织业', 'month_deviation_rate': 0.00456}
{'city': '台州', 'industry': '6.纺织服装、服饰业', 'month_deviation_rate': 0.00237}
{'city': '台州', 'industry': '7.皮革、毛皮、羽毛及其制品和制鞋业', 'month_deviation_rate': 0.00089}
{'city': '台州', 'industry': '8.木材加工和木、竹、藤、棕、草制品业', 'month_deviation_rate': 0.00067}
{'city': '台州', 'industry': '9.家具制造业', 'month_deviation_rate': 0.00493}
{'city': '台州', 'industry': '11.印刷和记录媒介复制业', 'month_deviation_rate': 0.00524}
{'city': '台州', 'industry': '12.文教、工美、体育和娱乐用品制造业', 'month_deviation_rate': 0.00593}
{'city': '台州', 'industry': '其中:体育用品制造', 'month_deviation_rate': 0.0122}
{'city': '台州', 'industry': '13.石油、煤炭及其他燃料加工业', 'month_deviation_rate': 0.00827}
{'city': '台州', 'industry': '14.化学原料和化学制品制造业', 'month_deviation_rate': -0.00164}
{'city': '台州', 'industry': '其中:肥料制造', 'month_deviation_rate': -0.01009}
{'city': '台州', 'industry': '其中:中成药生产', 'month_deviation_rate': 0.00083}
{'city': '台州', 'industry': '16.化学纤维制造业', 'month_deviation_rate': 0.00104}
{'city': '台州', 'industry': '17.橡胶和塑料制品业', 'month_deviation_rate': 0.00067}
{'city': '台州', 'industry': '其中:橡胶制品业', 'month_deviation_rate': 0.00118}
{'city': '台州', 'industry': '塑料制品业', 'month_deviation_rate': 0.00314}
{'city': '台州', 'industry': '18.非金属矿物制品业', 'month_deviation_rate': -0.00198}

@ -73,13 +73,13 @@ def to_data(file_dir, excel):
# 使用GPU
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
train_x.to(device)
train_y.to(device)
# 转为pytorch的tensor对象
train_x = torch.from_numpy(train_x)
train_y = torch.from_numpy(train_y)
train_x = train_x.cuda()
train_y = train_y.cuda()
model = LSTM_Regression(DAYS_FOR_TRAIN, 32, output_size=1, num_layers=2) # 导入模型并设置模型的参数输入输出层、隐藏层等
# 使用GPU
@ -93,7 +93,7 @@ def to_data(file_dir, excel):
optimizer = torch.optim.Adam(model.parameters(), lr=0.001, betas=(0.9, 0.999), eps=1e-08, weight_decay=0)
a = time.time()
print("行业加载时间", a-c)
print(industry, "行业加载时间", a - c)
for i in range(1200):
out = model(train_x)
loss = loss_function(out, train_y)
@ -113,14 +113,13 @@ def to_data(file_dir, excel):
model = model.eval() # 转换成测试模式
# model.load_state_dict(torch.load(os.path.join(model_save_dir,model_file))) # 读取参数
dataset_x = dataset_x.reshape(-1, 1, DAYS_FOR_TRAIN) # (seq_size, batch_size, feature_size)
dataset_x = torch.from_numpy(dataset_x)
dataset_x.to(device)
dataset_x = dataset_x.cuda()
pred_test = model(dataset_x) # 全量训练集
# 模型输出 (seq_size, batch_size, output_size)
pred_test = pred_test.cpu()
pred_test = pred_test.view(-1).data.numpy()
pred_test = np.concatenate((np.zeros(DAYS_FOR_TRAIN), pred_test))
assert len(pred_test) == len(df)
@ -143,16 +142,14 @@ def to_data(file_dir, excel):
print("========")
if __name__ == '__main__':
file_dir = r'./浙江所有地市133行业数据'
# p = Pool(1)
# for excel in os.listdir((file_dir)):
# p.apply_async(func=to_data, args=(file_dir, excel))
# p.close()
# p.join()
p = Pool(6)
for excel in os.listdir((file_dir)):
to_data(file_dir, excel)
p.apply_async(func=to_data, args=(file_dir, excel))
p.close()
p.join()
# for excel in os.listdir((file_dir)):
# to_data(file_dir, excel)

Loading…
Cancel
Save