You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
610 lines
76 KiB
610 lines
76 KiB
|
|
|
|
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
|
|
|
|
|
|
<meta http-equiv="X-UA-Compatible" content="IE=emulateIE7" />
|
|
<title>Coverage for LSTMPredictStock\run.py: 81%</title>
|
|
<link rel="stylesheet" href="style.css" type="text/css">
|
|
|
|
<script type="text/javascript" src="jquery.min.js"></script>
|
|
<script type="text/javascript" src="jquery.hotkeys.js"></script>
|
|
<script type="text/javascript" src="jquery.isonscreen.js"></script>
|
|
<script type="text/javascript" src="coverage_html.js"></script>
|
|
<script type="text/javascript">
|
|
jQuery(document).ready(coverage.pyfile_ready);
|
|
</script>
|
|
</head>
|
|
<body class="pyfile">
|
|
|
|
<div id="header">
|
|
<div class="content">
|
|
<h1>Coverage for <b>LSTMPredictStock\run.py</b> :
|
|
<span class="pc_cov">81%</span>
|
|
</h1>
|
|
|
|
<img id="keyboard_icon" src="keybd_closed.png" alt="Show keyboard shortcuts" />
|
|
|
|
<h2 class="stats">
|
|
126 statements
|
|
<span class="run hide_run shortkey_r button_toggle_run">102 run</span>
|
|
<span class="mis shortkey_m button_toggle_mis">24 missing</span>
|
|
<span class="exc shortkey_x button_toggle_exc">0 excluded</span>
|
|
|
|
|
|
</h2>
|
|
</div>
|
|
</div>
|
|
|
|
<div class="help_panel">
|
|
<img id="panel_icon" src="keybd_open.png" alt="Hide keyboard shortcuts" />
|
|
<p class="legend">Hot-keys on this page</p>
|
|
<div>
|
|
<p class="keyhelp">
|
|
<span class="key">r</span>
|
|
<span class="key">m</span>
|
|
<span class="key">x</span>
|
|
<span class="key">p</span> toggle line displays
|
|
</p>
|
|
<p class="keyhelp">
|
|
<span class="key">j</span>
|
|
<span class="key">k</span> next/prev highlighted chunk
|
|
</p>
|
|
<p class="keyhelp">
|
|
<span class="key">0</span> (zero) top of page
|
|
</p>
|
|
<p class="keyhelp">
|
|
<span class="key">1</span> (one) first highlighted chunk
|
|
</p>
|
|
</div>
|
|
</div>
|
|
|
|
<div id="source">
|
|
<table>
|
|
<tr>
|
|
<td class="linenos">
|
|
<p id="n1" class="stm run hide_run"><a href="#n1">1</a></p>
|
|
<p id="n2" class="pln"><a href="#n2">2</a></p>
|
|
<p id="n3" class="stm run hide_run"><a href="#n3">3</a></p>
|
|
<p id="n4" class="stm run hide_run"><a href="#n4">4</a></p>
|
|
<p id="n5" class="stm run hide_run"><a href="#n5">5</a></p>
|
|
<p id="n6" class="stm run hide_run"><a href="#n6">6</a></p>
|
|
<p id="n7" class="pln"><a href="#n7">7</a></p>
|
|
<p id="n8" class="stm run hide_run"><a href="#n8">8</a></p>
|
|
<p id="n9" class="stm run hide_run"><a href="#n9">9</a></p>
|
|
<p id="n10" class="stm run hide_run"><a href="#n10">10</a></p>
|
|
<p id="n11" class="stm run hide_run"><a href="#n11">11</a></p>
|
|
<p id="n12" class="stm run hide_run"><a href="#n12">12</a></p>
|
|
<p id="n13" class="stm run hide_run"><a href="#n13">13</a></p>
|
|
<p id="n14" class="stm run hide_run"><a href="#n14">14</a></p>
|
|
<p id="n15" class="stm run hide_run"><a href="#n15">15</a></p>
|
|
<p id="n16" class="stm run hide_run"><a href="#n16">16</a></p>
|
|
<p id="n17" class="stm run hide_run"><a href="#n17">17</a></p>
|
|
<p id="n18" class="stm run hide_run"><a href="#n18">18</a></p>
|
|
<p id="n19" class="pln"><a href="#n19">19</a></p>
|
|
<p id="n20" class="pln"><a href="#n20">20</a></p>
|
|
<p id="n21" class="stm run hide_run"><a href="#n21">21</a></p>
|
|
<p id="n22" class="stm mis"><a href="#n22">22</a></p>
|
|
<p id="n23" class="stm mis"><a href="#n23">23</a></p>
|
|
<p id="n24" class="stm mis"><a href="#n24">24</a></p>
|
|
<p id="n25" class="stm mis"><a href="#n25">25</a></p>
|
|
<p id="n26" class="stm mis"><a href="#n26">26</a></p>
|
|
<p id="n27" class="stm mis"><a href="#n27">27</a></p>
|
|
<p id="n28" class="pln"><a href="#n28">28</a></p>
|
|
<p id="n29" class="pln"><a href="#n29">29</a></p>
|
|
<p id="n30" class="pln"><a href="#n30">30</a></p>
|
|
<p id="n31" class="stm run hide_run"><a href="#n31">31</a></p>
|
|
<p id="n32" class="stm mis"><a href="#n32">32</a></p>
|
|
<p id="n33" class="stm mis"><a href="#n33">33</a></p>
|
|
<p id="n34" class="stm mis"><a href="#n34">34</a></p>
|
|
<p id="n35" class="pln"><a href="#n35">35</a></p>
|
|
<p id="n36" class="stm mis"><a href="#n36">36</a></p>
|
|
<p id="n37" class="stm mis"><a href="#n37">37</a></p>
|
|
<p id="n38" class="stm mis"><a href="#n38">38</a></p>
|
|
<p id="n39" class="stm mis"><a href="#n39">39</a></p>
|
|
<p id="n40" class="stm mis"><a href="#n40">40</a></p>
|
|
<p id="n41" class="pln"><a href="#n41">41</a></p>
|
|
<p id="n42" class="pln"><a href="#n42">42</a></p>
|
|
<p id="n43" class="pln"><a href="#n43">43</a></p>
|
|
<p id="n44" class="stm run hide_run"><a href="#n44">44</a></p>
|
|
<p id="n45" class="pln"><a href="#n45">45</a></p>
|
|
<p id="n46" class="pln"><a href="#n46">46</a></p>
|
|
<p id="n47" class="pln"><a href="#n47">47</a></p>
|
|
<p id="n48" class="pln"><a href="#n48">48</a></p>
|
|
<p id="n49" class="stm run hide_run"><a href="#n49">49</a></p>
|
|
<p id="n50" class="stm run hide_run"><a href="#n50">50</a></p>
|
|
<p id="n51" class="stm mis"><a href="#n51">51</a></p>
|
|
<p id="n52" class="pln"><a href="#n52">52</a></p>
|
|
<p id="n53" class="stm run hide_run"><a href="#n53">53</a></p>
|
|
<p id="n54" class="stm run hide_run"><a href="#n54">54</a></p>
|
|
<p id="n55" class="stm run hide_run"><a href="#n55">55</a></p>
|
|
<p id="n56" class="pln"><a href="#n56">56</a></p>
|
|
<p id="n57" class="stm run hide_run"><a href="#n57">57</a></p>
|
|
<p id="n58" class="pln"><a href="#n58">58</a></p>
|
|
<p id="n59" class="pln"><a href="#n59">59</a></p>
|
|
<p id="n60" class="pln"><a href="#n60">60</a></p>
|
|
<p id="n61" class="pln"><a href="#n61">61</a></p>
|
|
<p id="n62" class="pln"><a href="#n62">62</a></p>
|
|
<p id="n63" class="stm run hide_run"><a href="#n63">63</a></p>
|
|
<p id="n64" class="stm run hide_run"><a href="#n64">64</a></p>
|
|
<p id="n65" class="pln"><a href="#n65">65</a></p>
|
|
<p id="n66" class="pln"><a href="#n66">66</a></p>
|
|
<p id="n67" class="pln"><a href="#n67">67</a></p>
|
|
<p id="n68" class="pln"><a href="#n68">68</a></p>
|
|
<p id="n69" class="pln"><a href="#n69">69</a></p>
|
|
<p id="n70" class="pln"><a href="#n70">70</a></p>
|
|
<p id="n71" class="pln"><a href="#n71">71</a></p>
|
|
<p id="n72" class="pln"><a href="#n72">72</a></p>
|
|
<p id="n73" class="pln"><a href="#n73">73</a></p>
|
|
<p id="n74" class="pln"><a href="#n74">74</a></p>
|
|
<p id="n75" class="pln"><a href="#n75">75</a></p>
|
|
<p id="n76" class="pln"><a href="#n76">76</a></p>
|
|
<p id="n77" class="pln"><a href="#n77">77</a></p>
|
|
<p id="n78" class="stm run hide_run"><a href="#n78">78</a></p>
|
|
<p id="n79" class="pln"><a href="#n79">79</a></p>
|
|
<p id="n80" class="stm run hide_run"><a href="#n80">80</a></p>
|
|
<p id="n81" class="pln"><a href="#n81">81</a></p>
|
|
<p id="n82" class="pln"><a href="#n82">82</a></p>
|
|
<p id="n83" class="pln"><a href="#n83">83</a></p>
|
|
<p id="n84" class="pln"><a href="#n84">84</a></p>
|
|
<p id="n85" class="pln"><a href="#n85">85</a></p>
|
|
<p id="n86" class="pln"><a href="#n86">86</a></p>
|
|
<p id="n87" class="pln"><a href="#n87">87</a></p>
|
|
<p id="n88" class="pln"><a href="#n88">88</a></p>
|
|
<p id="n89" class="pln"><a href="#n89">89</a></p>
|
|
<p id="n90" class="pln"><a href="#n90">90</a></p>
|
|
<p id="n91" class="pln"><a href="#n91">91</a></p>
|
|
<p id="n92" class="pln"><a href="#n92">92</a></p>
|
|
<p id="n93" class="pln"><a href="#n93">93</a></p>
|
|
<p id="n94" class="stm run hide_run"><a href="#n94">94</a></p>
|
|
<p id="n95" class="stm mis"><a href="#n95">95</a></p>
|
|
<p id="n96" class="pln"><a href="#n96">96</a></p>
|
|
<p id="n97" class="pln"><a href="#n97">97</a></p>
|
|
<p id="n98" class="pln"><a href="#n98">98</a></p>
|
|
<p id="n99" class="pln"><a href="#n99">99</a></p>
|
|
<p id="n100" class="stm mis"><a href="#n100">100</a></p>
|
|
<p id="n101" class="pln"><a href="#n101">101</a></p>
|
|
<p id="n102" class="stm mis"><a href="#n102">102</a></p>
|
|
<p id="n103" class="pln"><a href="#n103">103</a></p>
|
|
<p id="n104" class="pln"><a href="#n104">104</a></p>
|
|
<p id="n105" class="pln"><a href="#n105">105</a></p>
|
|
<p id="n106" class="pln"><a href="#n106">106</a></p>
|
|
<p id="n107" class="stm run hide_run"><a href="#n107">107</a></p>
|
|
<p id="n108" class="pln"><a href="#n108">108</a></p>
|
|
<p id="n109" class="pln"><a href="#n109">109</a></p>
|
|
<p id="n110" class="pln"><a href="#n110">110</a></p>
|
|
<p id="n111" class="stm run hide_run"><a href="#n111">111</a></p>
|
|
<p id="n112" class="stm run hide_run"><a href="#n112">112</a></p>
|
|
<p id="n113" class="stm run hide_run"><a href="#n113">113</a></p>
|
|
<p id="n114" class="pln"><a href="#n114">114</a></p>
|
|
<p id="n115" class="pln"><a href="#n115">115</a></p>
|
|
<p id="n116" class="pln"><a href="#n116">116</a></p>
|
|
<p id="n117" class="pln"><a href="#n117">117</a></p>
|
|
<p id="n118" class="pln"><a href="#n118">118</a></p>
|
|
<p id="n119" class="stm run hide_run"><a href="#n119">119</a></p>
|
|
<p id="n120" class="stm run hide_run"><a href="#n120">120</a></p>
|
|
<p id="n121" class="stm run hide_run"><a href="#n121">121</a></p>
|
|
<p id="n122" class="stm run hide_run"><a href="#n122">122</a></p>
|
|
<p id="n123" class="pln"><a href="#n123">123</a></p>
|
|
<p id="n124" class="pln"><a href="#n124">124</a></p>
|
|
<p id="n125" class="stm run hide_run"><a href="#n125">125</a></p>
|
|
<p id="n126" class="stm run hide_run"><a href="#n126">126</a></p>
|
|
<p id="n127" class="stm run hide_run"><a href="#n127">127</a></p>
|
|
<p id="n128" class="pln"><a href="#n128">128</a></p>
|
|
<p id="n129" class="stm mis"><a href="#n129">129</a></p>
|
|
<p id="n130" class="pln"><a href="#n130">130</a></p>
|
|
<p id="n131" class="stm run hide_run"><a href="#n131">131</a></p>
|
|
<p id="n132" class="pln"><a href="#n132">132</a></p>
|
|
<p id="n133" class="pln"><a href="#n133">133</a></p>
|
|
<p id="n134" class="pln"><a href="#n134">134</a></p>
|
|
<p id="n135" class="pln"><a href="#n135">135</a></p>
|
|
<p id="n136" class="stm run hide_run"><a href="#n136">136</a></p>
|
|
<p id="n137" class="stm run hide_run"><a href="#n137">137</a></p>
|
|
<p id="n138" class="stm run hide_run"><a href="#n138">138</a></p>
|
|
<p id="n139" class="stm mis"><a href="#n139">139</a></p>
|
|
<p id="n140" class="pln"><a href="#n140">140</a></p>
|
|
<p id="n141" class="stm run hide_run"><a href="#n141">141</a></p>
|
|
<p id="n142" class="stm run hide_run"><a href="#n142">142</a></p>
|
|
<p id="n143" class="pln"><a href="#n143">143</a></p>
|
|
<p id="n144" class="stm run hide_run"><a href="#n144">144</a></p>
|
|
<p id="n145" class="pln"><a href="#n145">145</a></p>
|
|
<p id="n146" class="pln"><a href="#n146">146</a></p>
|
|
<p id="n147" class="pln"><a href="#n147">147</a></p>
|
|
<p id="n148" class="stm run hide_run"><a href="#n148">148</a></p>
|
|
<p id="n149" class="stm run hide_run"><a href="#n149">149</a></p>
|
|
<p id="n150" class="pln"><a href="#n150">150</a></p>
|
|
<p id="n151" class="pln"><a href="#n151">151</a></p>
|
|
<p id="n152" class="pln"><a href="#n152">152</a></p>
|
|
<p id="n153" class="pln"><a href="#n153">153</a></p>
|
|
<p id="n154" class="stm run hide_run"><a href="#n154">154</a></p>
|
|
<p id="n155" class="pln"><a href="#n155">155</a></p>
|
|
<p id="n156" class="stm run hide_run"><a href="#n156">156</a></p>
|
|
<p id="n157" class="stm run hide_run"><a href="#n157">157</a></p>
|
|
<p id="n158" class="stm run hide_run"><a href="#n158">158</a></p>
|
|
<p id="n159" class="pln"><a href="#n159">159</a></p>
|
|
<p id="n160" class="pln"><a href="#n160">160</a></p>
|
|
<p id="n161" class="stm run hide_run"><a href="#n161">161</a></p>
|
|
<p id="n162" class="stm mis"><a href="#n162">162</a></p>
|
|
<p id="n163" class="pln"><a href="#n163">163</a></p>
|
|
<p id="n164" class="pln"><a href="#n164">164</a></p>
|
|
<p id="n165" class="stm run hide_run"><a href="#n165">165</a></p>
|
|
<p id="n166" class="stm mis"><a href="#n166">166</a></p>
|
|
<p id="n167" class="stm mis"><a href="#n167">167</a></p>
|
|
<p id="n168" class="pln"><a href="#n168">168</a></p>
|
|
<p id="n169" class="stm mis"><a href="#n169">169</a></p>
|
|
<p id="n170" class="pln"><a href="#n170">170</a></p>
|
|
<p id="n171" class="stm run hide_run"><a href="#n171">171</a></p>
|
|
<p id="n172" class="pln"><a href="#n172">172</a></p>
|
|
<p id="n173" class="stm run hide_run"><a href="#n173">173</a></p>
|
|
<p id="n174" class="stm run hide_run"><a href="#n174">174</a></p>
|
|
<p id="n175" class="stm run hide_run"><a href="#n175">175</a></p>
|
|
<p id="n176" class="stm run hide_run"><a href="#n176">176</a></p>
|
|
<p id="n177" class="stm run hide_run"><a href="#n177">177</a></p>
|
|
<p id="n178" class="pln"><a href="#n178">178</a></p>
|
|
<p id="n179" class="stm run hide_run"><a href="#n179">179</a></p>
|
|
<p id="n180" class="stm run hide_run"><a href="#n180">180</a></p>
|
|
<p id="n181" class="stm run hide_run"><a href="#n181">181</a></p>
|
|
<p id="n182" class="stm run hide_run"><a href="#n182">182</a></p>
|
|
<p id="n183" class="stm run hide_run"><a href="#n183">183</a></p>
|
|
<p id="n184" class="stm run hide_run"><a href="#n184">184</a></p>
|
|
<p id="n185" class="stm run hide_run"><a href="#n185">185</a></p>
|
|
<p id="n186" class="stm run hide_run"><a href="#n186">186</a></p>
|
|
<p id="n187" class="pln"><a href="#n187">187</a></p>
|
|
<p id="n188" class="stm run hide_run"><a href="#n188">188</a></p>
|
|
<p id="n189" class="pln"><a href="#n189">189</a></p>
|
|
<p id="n190" class="pln"><a href="#n190">190</a></p>
|
|
<p id="n191" class="pln"><a href="#n191">191</a></p>
|
|
<p id="n192" class="pln"><a href="#n192">192</a></p>
|
|
<p id="n193" class="pln"><a href="#n193">193</a></p>
|
|
<p id="n194" class="pln"><a href="#n194">194</a></p>
|
|
<p id="n195" class="pln"><a href="#n195">195</a></p>
|
|
<p id="n196" class="pln"><a href="#n196">196</a></p>
|
|
<p id="n197" class="pln"><a href="#n197">197</a></p>
|
|
<p id="n198" class="pln"><a href="#n198">198</a></p>
|
|
<p id="n199" class="pln"><a href="#n199">199</a></p>
|
|
<p id="n200" class="pln"><a href="#n200">200</a></p>
|
|
<p id="n201" class="pln"><a href="#n201">201</a></p>
|
|
<p id="n202" class="pln"><a href="#n202">202</a></p>
|
|
<p id="n203" class="pln"><a href="#n203">203</a></p>
|
|
<p id="n204" class="pln"><a href="#n204">204</a></p>
|
|
<p id="n205" class="pln"><a href="#n205">205</a></p>
|
|
<p id="n206" class="pln"><a href="#n206">206</a></p>
|
|
<p id="n207" class="pln"><a href="#n207">207</a></p>
|
|
<p id="n208" class="pln"><a href="#n208">208</a></p>
|
|
<p id="n209" class="pln"><a href="#n209">209</a></p>
|
|
<p id="n210" class="pln"><a href="#n210">210</a></p>
|
|
<p id="n211" class="pln"><a href="#n211">211</a></p>
|
|
<p id="n212" class="stm run hide_run"><a href="#n212">212</a></p>
|
|
<p id="n213" class="stm run hide_run"><a href="#n213">213</a></p>
|
|
<p id="n214" class="stm run hide_run"><a href="#n214">214</a></p>
|
|
<p id="n215" class="stm run hide_run"><a href="#n215">215</a></p>
|
|
<p id="n216" class="stm run hide_run"><a href="#n216">216</a></p>
|
|
<p id="n217" class="stm run hide_run"><a href="#n217">217</a></p>
|
|
<p id="n218" class="stm run hide_run"><a href="#n218">218</a></p>
|
|
<p id="n219" class="stm run hide_run"><a href="#n219">219</a></p>
|
|
<p id="n220" class="pln"><a href="#n220">220</a></p>
|
|
<p id="n221" class="pln"><a href="#n221">221</a></p>
|
|
<p id="n222" class="stm run hide_run"><a href="#n222">222</a></p>
|
|
<p id="n223" class="stm run hide_run"><a href="#n223">223</a></p>
|
|
<p id="n224" class="stm run hide_run"><a href="#n224">224</a></p>
|
|
<p id="n225" class="stm run hide_run"><a href="#n225">225</a></p>
|
|
<p id="n226" class="stm run hide_run"><a href="#n226">226</a></p>
|
|
<p id="n227" class="stm run hide_run"><a href="#n227">227</a></p>
|
|
<p id="n228" class="pln"><a href="#n228">228</a></p>
|
|
<p id="n229" class="stm run hide_run"><a href="#n229">229</a></p>
|
|
<p id="n230" class="pln"><a href="#n230">230</a></p>
|
|
<p id="n231" class="pln"><a href="#n231">231</a></p>
|
|
<p id="n232" class="stm run hide_run"><a href="#n232">232</a></p>
|
|
<p id="n233" class="stm run hide_run"><a href="#n233">233</a></p>
|
|
<p id="n234" class="stm run hide_run"><a href="#n234">234</a></p>
|
|
<p id="n235" class="stm run hide_run"><a href="#n235">235</a></p>
|
|
<p id="n236" class="stm run hide_run"><a href="#n236">236</a></p>
|
|
<p id="n237" class="stm run hide_run"><a href="#n237">237</a></p>
|
|
<p id="n238" class="stm run hide_run"><a href="#n238">238</a></p>
|
|
<p id="n239" class="pln"><a href="#n239">239</a></p>
|
|
<p id="n240" class="stm run hide_run"><a href="#n240">240</a></p>
|
|
<p id="n241" class="pln"><a href="#n241">241</a></p>
|
|
<p id="n242" class="pln"><a href="#n242">242</a></p>
|
|
<p id="n243" class="stm run hide_run"><a href="#n243">243</a></p>
|
|
<p id="n244" class="stm run hide_run"><a href="#n244">244</a></p>
|
|
<p id="n245" class="stm run hide_run"><a href="#n245">245</a></p>
|
|
<p id="n246" class="pln"><a href="#n246">246</a></p>
|
|
<p id="n247" class="pln"><a href="#n247">247</a></p>
|
|
<p id="n248" class="stm run hide_run"><a href="#n248">248</a></p>
|
|
<p id="n249" class="stm run hide_run"><a href="#n249">249</a></p>
|
|
<p id="n250" class="stm run hide_run"><a href="#n250">250</a></p>
|
|
<p id="n251" class="pln"><a href="#n251">251</a></p>
|
|
<p id="n252" class="pln"><a href="#n252">252</a></p>
|
|
<p id="n253" class="stm run hide_run"><a href="#n253">253</a></p>
|
|
<p id="n254" class="stm run hide_run"><a href="#n254">254</a></p>
|
|
<p id="n255" class="pln"><a href="#n255">255</a></p>
|
|
<p id="n256" class="pln"><a href="#n256">256</a></p>
|
|
<p id="n257" class="pln"><a href="#n257">257</a></p>
|
|
<p id="n258" class="pln"><a href="#n258">258</a></p>
|
|
<p id="n259" class="pln"><a href="#n259">259</a></p>
|
|
<p id="n260" class="pln"><a href="#n260">260</a></p>
|
|
|
|
</td>
|
|
<td class="text">
|
|
<p id="t1" class="stm run hide_run"><span class="key">import</span> <span class="nam">keras</span><span class="strut"> </span></p>
|
|
<p id="t2" class="pln"><span class="strut"> </span></p>
|
|
<p id="t3" class="stm run hide_run"><span class="nam">__author__</span> <span class="op">=</span> <span class="str">"Jakob Aungiers"</span><span class="strut"> </span></p>
|
|
<p id="t4" class="stm run hide_run"><span class="nam">__copyright__</span> <span class="op">=</span> <span class="str">"Jakob Aungiers 2018"</span><span class="strut"> </span></p>
|
|
<p id="t5" class="stm run hide_run"><span class="nam">__version__</span> <span class="op">=</span> <span class="str">"2.0.0"</span><span class="strut"> </span></p>
|
|
<p id="t6" class="stm run hide_run"><span class="nam">__license__</span> <span class="op">=</span> <span class="str">"MIT"</span><span class="strut"> </span></p>
|
|
<p id="t7" class="pln"><span class="strut"> </span></p>
|
|
<p id="t8" class="stm run hide_run"><span class="key">import</span> <span class="nam">os</span><span class="strut"> </span></p>
|
|
<p id="t9" class="stm run hide_run"><span class="key">import</span> <span class="nam">json</span><span class="strut"> </span></p>
|
|
<p id="t10" class="stm run hide_run"><span class="key">import</span> <span class="nam">numpy</span> <span class="key">as</span> <span class="nam">np</span><span class="strut"> </span></p>
|
|
<p id="t11" class="stm run hide_run"><span class="key">import</span> <span class="nam">math</span><span class="strut"> </span></p>
|
|
<p id="t12" class="stm run hide_run"><span class="key">import</span> <span class="nam">pandas</span> <span class="key">as</span> <span class="nam">pd</span><span class="strut"> </span></p>
|
|
<p id="t13" class="stm run hide_run"><span class="key">import</span> <span class="nam">matplotlib</span><span class="op">.</span><span class="nam">pyplot</span> <span class="key">as</span> <span class="nam">plt</span><span class="strut"> </span></p>
|
|
<p id="t14" class="stm run hide_run"><span class="key">from</span> <span class="nam">LSTMPredictStock</span><span class="op">.</span><span class="nam">core</span><span class="op">.</span><span class="nam">data_processor</span> <span class="key">import</span> <span class="nam">DataLoader</span><span class="strut"> </span></p>
|
|
<p id="t15" class="stm run hide_run"><span class="key">from</span> <span class="nam">LSTMPredictStock</span><span class="op">.</span><span class="nam">core</span><span class="op">.</span><span class="nam">model</span> <span class="key">import</span> <span class="nam">Model</span><span class="strut"> </span></p>
|
|
<p id="t16" class="stm run hide_run"><span class="key">from</span> <span class="nam">datetime</span> <span class="key">import</span> <span class="nam">datetime</span><span class="op">,</span><span class="nam">timedelta</span><span class="strut"> </span></p>
|
|
<p id="t17" class="stm run hide_run"><span class="key">from</span> <span class="nam">LSTMPredictStock</span><span class="op">.</span><span class="nam">core</span><span class="op">.</span><span class="nam">get_domestic_hist_stock</span> <span class="key">import</span> <span class="nam">get_all_last_data</span><span class="strut"> </span></p>
|
|
<p id="t18" class="stm run hide_run"><span class="key">from</span> <span class="nam">LSTMPredictStock</span><span class="op">.</span><span class="nam">core</span><span class="op">.</span><span class="nam">get_domestic_hist_stock</span> <span class="key">import</span> <span class="nam">get_single_last_data</span><span class="strut"> </span></p>
|
|
<p id="t19" class="pln"><span class="strut"> </span></p>
|
|
<p id="t20" class="pln"><span class="strut"> </span></p>
|
|
<p id="t21" class="stm run hide_run"><span class="key">def</span> <span class="nam">plot_results</span><span class="op">(</span><span class="nam">predicted_data</span><span class="op">,</span> <span class="nam">true_data</span><span class="op">)</span><span class="op">:</span> <span class="com"># predicted_data与true_data：同长度一维数组</span><span class="strut"> </span></p>
|
|
<p id="t22" class="stm mis"> <span class="nam">fig</span> <span class="op">=</span> <span class="nam">plt</span><span class="op">.</span><span class="nam">figure</span><span class="op">(</span><span class="nam">facecolor</span><span class="op">=</span><span class="str">'white'</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t23" class="stm mis"> <span class="nam">ax</span> <span class="op">=</span> <span class="nam">fig</span><span class="op">.</span><span class="nam">add_subplot</span><span class="op">(</span><span class="num">111</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t24" class="stm mis"> <span class="nam">ax</span><span class="op">.</span><span class="nam">plot</span><span class="op">(</span><span class="nam">true_data</span><span class="op">,</span> <span class="nam">label</span><span class="op">=</span><span class="str">'True Data'</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t25" class="stm mis"> <span class="nam">plt</span><span class="op">.</span><span class="nam">plot</span><span class="op">(</span><span class="nam">predicted_data</span><span class="op">,</span> <span class="nam">label</span><span class="op">=</span><span class="str">'Prediction'</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t26" class="stm mis"> <span class="nam">plt</span><span class="op">.</span><span class="nam">legend</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t27" class="stm mis"> <span class="nam">plt</span><span class="op">.</span><span class="nam">show</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t28" class="pln"><span class="strut"> </span></p>
|
|
<p id="t29" class="pln"><span class="strut"> </span></p>
|
|
<p id="t30" class="pln"><span class="com"># predicted_data每个元素的长度必须为prediction_len</span><span class="strut"> </span></p>
|
|
<p id="t31" class="stm run hide_run"><span class="key">def</span> <span class="nam">plot_results_multiple</span><span class="op">(</span><span class="nam">predicted_data</span><span class="op">,</span> <span class="nam">true_data</span><span class="op">,</span> <span class="nam">prediction_len</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t32" class="stm mis"> <span class="nam">fig</span> <span class="op">=</span> <span class="nam">plt</span><span class="op">.</span><span class="nam">figure</span><span class="op">(</span><span class="nam">facecolor</span><span class="op">=</span><span class="str">'white'</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t33" class="stm mis"> <span class="nam">ax</span> <span class="op">=</span> <span class="nam">fig</span><span class="op">.</span><span class="nam">add_subplot</span><span class="op">(</span><span class="num">111</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t34" class="stm mis"> <span class="nam">ax</span><span class="op">.</span><span class="nam">plot</span><span class="op">(</span><span class="nam">true_data</span><span class="op">,</span> <span class="nam">label</span><span class="op">=</span><span class="str">'True Data'</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t35" class="pln"> <span class="com"># Pad the list of predictions to shift it in the graph to it's correct start</span><span class="strut"> </span></p>
|
|
<p id="t36" class="stm mis"> <span class="key">for</span> <span class="nam">i</span><span class="op">,</span> <span class="nam">data</span> <span class="key">in</span> <span class="nam">enumerate</span><span class="op">(</span><span class="nam">predicted_data</span><span class="op">)</span><span class="op">:</span> <span class="com"># data为一维数组，长度为prediction_len。predicted_data：二维数组，每个元素为list</span><span class="strut"> </span></p>
|
|
<p id="t37" class="stm mis"> <span class="nam">padding</span> <span class="op">=</span> <span class="op">[</span><span class="key">None</span> <span class="key">for</span> <span class="nam">p</span> <span class="key">in</span> <span class="nam">range</span><span class="op">(</span><span class="nam">i</span> <span class="op">*</span> <span class="nam">prediction_len</span><span class="op">)</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t38" class="stm mis"> <span class="nam">plt</span><span class="op">.</span><span class="nam">plot</span><span class="op">(</span><span class="nam">padding</span> <span class="op">+</span> <span class="nam">data</span><span class="op">,</span> <span class="nam">label</span><span class="op">=</span><span class="str">'Prediction'</span><span class="op">)</span> <span class="com"># padding + data：list拼接操作</span><span class="strut"> </span></p>
|
|
<p id="t39" class="stm mis"> <span class="nam">plt</span><span class="op">.</span><span class="nam">legend</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t40" class="stm mis"> <span class="nam">plt</span><span class="op">.</span><span class="nam">show</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t41" class="pln"><span class="strut"> </span></p>
|
|
<p id="t42" class="pln"><span class="strut"> </span></p>
|
|
<p id="t43" class="pln"><span class="com"># 只用于训练模型，但同时可根据参数进行模型的评估</span><span class="strut"> </span></p>
|
|
<p id="t44" class="stm run hide_run"><span class="key">def</span> <span class="nam">train_model</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">,</span> <span class="nam">predict</span><span class="op">=</span><span class="key">False</span><span class="op">)</span><span class="op">:</span> <span class="com"># 训练指定股票代码的模型</span><span class="strut"> </span></p>
|
|
<p id="t45" class="pln"> <span class="str">'''</span><span class="strut"> </span></p>
|
|
<p id="t46" class="pln"><span class="str"> 训练并保存模型，同时根据测试数据对模型进行评估（绘图方式）</span><span class="strut"> </span></p>
|
|
<p id="t47" class="pln"><span class="str"> '''</span><span class="strut"> </span></p>
|
|
<p id="t48" class="pln"><span class="strut"> </span></p>
|
|
<p id="t49" class="stm run hide_run"> <span class="nam">configs</span> <span class="op">=</span> <span class="nam">json</span><span class="op">.</span><span class="nam">load</span><span class="op">(</span><span class="nam">open</span><span class="op">(</span><span class="nam">get_config_path</span><span class="op">(</span><span class="op">)</span><span class="op">,</span> <span class="str">'r'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t50" class="stm run hide_run"> <span class="key">if</span> <span class="key">not</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">exists</span><span class="op">(</span><span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="op">,</span><span class="nam">configs</span><span class="op">[</span><span class="str">'model'</span><span class="op">]</span><span class="op">[</span><span class="str">'save_dir'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t51" class="stm mis"> <span class="nam">os</span><span class="op">.</span><span class="nam">makedirs</span><span class="op">(</span><span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="op">,</span><span class="nam">configs</span><span class="op">[</span><span class="str">'model'</span><span class="op">]</span><span class="op">[</span><span class="str">'save_dir'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> <span class="com"># 创建保存模型的目录</span><span class="strut"> </span></p>
|
|
<p id="t52" class="pln"><span class="strut"> </span></p>
|
|
<p id="t53" class="stm run hide_run"> <span class="nam">split</span> <span class="op">=</span> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'train_test_split'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t54" class="stm run hide_run"> <span class="key">if</span> <span class="key">not</span> <span class="nam">predict</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t55" class="stm run hide_run"> <span class="nam">split</span> <span class="op">=</span> <span class="num">1</span> <span class="com"># 若不评估模型准确度，则将全部历史数据用于训练</span><span class="strut"> </span></p>
|
|
<p id="t56" class="pln"><span class="strut"> </span></p>
|
|
<p id="t57" class="stm run hide_run"> <span class="nam">data</span> <span class="op">=</span> <span class="nam">DataLoader</span><span class="op">(</span> <span class="com"># 从本地加载训练和测试数据</span><span class="strut"> </span></p>
|
|
<p id="t58" class="pln"> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="op">,</span><span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="str">'data'</span><span class="op">,</span> <span class="nam">stock_code</span> <span class="op">+</span> <span class="str">".csv"</span><span class="op">)</span><span class="op">)</span><span class="op">,</span> <span class="com"># configs['data']['filename']</span><span class="strut"> </span></p>
|
|
<p id="t59" class="pln"> <span class="nam">split</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t60" class="pln"> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'columns'</span><span class="op">]</span> <span class="com"># 选择某些列的数据进行训练</span><span class="strut"> </span></p>
|
|
<p id="t61" class="pln"> <span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t62" class="pln"><span class="strut"> </span></p>
|
|
<p id="t63" class="stm run hide_run"> <span class="nam">model</span> <span class="op">=</span> <span class="nam">Model</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t64" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">build_model</span><span class="op">(</span><span class="nam">configs</span><span class="op">)</span> <span class="com"># 根据配置文件新建模型</span><span class="strut"> </span></p>
|
|
<p id="t65" class="pln"><span class="strut"> </span></p>
|
|
<p id="t66" class="pln"> <span class="str">'''</span><span class="strut"> </span></p>
|
|
<p id="t67" class="pln"><span class="str"> # in-memory training</span><span class="strut"> </span></p>
|
|
<p id="t68" class="pln"><span class="str"> model.train(</span><span class="strut"> </span></p>
|
|
<p id="t69" class="pln"><span class="str"> x,</span><span class="strut"> </span></p>
|
|
<p id="t70" class="pln"><span class="str"> y,</span><span class="strut"> </span></p>
|
|
<p id="t71" class="pln"><span class="str"> epochs = configs['training']['epochs'],</span><span class="strut"> </span></p>
|
|
<p id="t72" class="pln"><span class="str"> batch_size = configs['training']['batch_size'],</span><span class="strut"> </span></p>
|
|
<p id="t73" class="pln"><span class="str"> save_dir = configs['model']['save_dir']</span><span class="strut"> </span></p>
|
|
<p id="t74" class="pln"><span class="str"> )</span><span class="strut"> </span></p>
|
|
<p id="t75" class="pln"><span class="str"> '''</span><span class="strut"> </span></p>
|
|
<p id="t76" class="pln"> <span class="com"># 训练模型：</span><span class="strut"> </span></p>
|
|
<p id="t77" class="pln"> <span class="com"># out-of memory generative training</span><span class="strut"> </span></p>
|
|
<p id="t78" class="stm run hide_run"> <span class="nam">steps_per_epoch</span> <span class="op">=</span> <span class="nam">math</span><span class="op">.</span><span class="nam">ceil</span><span class="op">(</span><span class="strut"> </span></p>
|
|
<p id="t79" class="pln"> <span class="op">(</span><span class="nam">data</span><span class="op">.</span><span class="nam">len_train</span> <span class="op">-</span> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">)</span> <span class="op">/</span> <span class="nam">configs</span><span class="op">[</span><span class="str">'training'</span><span class="op">]</span><span class="op">[</span><span class="str">'batch_size'</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t80" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">train_generator</span><span class="op">(</span><span class="strut"> </span></p>
|
|
<p id="t81" class="pln"> <span class="nam">data_gen</span><span class="op">=</span><span class="nam">data</span><span class="op">.</span><span class="nam">generate_train_batch</span><span class="op">(</span><span class="strut"> </span></p>
|
|
<p id="t82" class="pln"> <span class="nam">seq_len</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t83" class="pln"> <span class="nam">batch_size</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'training'</span><span class="op">]</span><span class="op">[</span><span class="str">'batch_size'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t84" class="pln"> <span class="nam">normalise</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'normalise'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t85" class="pln"> <span class="op">)</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t86" class="pln"> <span class="nam">epochs</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'training'</span><span class="op">]</span><span class="op">[</span><span class="str">'epochs'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t87" class="pln"> <span class="nam">batch_size</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'training'</span><span class="op">]</span><span class="op">[</span><span class="str">'batch_size'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t88" class="pln"> <span class="nam">steps_per_epoch</span><span class="op">=</span><span class="nam">steps_per_epoch</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t89" class="pln"> <span class="nam">save_dir</span><span class="op">=</span><span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="op">,</span><span class="nam">configs</span><span class="op">[</span><span class="str">'model'</span><span class="op">]</span><span class="op">[</span><span class="str">'save_dir'</span><span class="op">]</span><span class="op">)</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t90" class="pln"> <span class="nam">save_name</span><span class="op">=</span><span class="nam">stock_code</span><span class="strut"> </span></p>
|
|
<p id="t91" class="pln"> <span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t92" class="pln"><span class="strut"> </span></p>
|
|
<p id="t93" class="pln"> <span class="com"># 预测</span><span class="strut"> </span></p>
|
|
<p id="t94" class="stm run hide_run"> <span class="key">if</span> <span class="nam">predict</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t95" class="stm mis"> <span class="nam">x_test</span><span class="op">,</span> <span class="nam">y_test</span> <span class="op">=</span> <span class="nam">data</span><span class="op">.</span><span class="nam">get_test_data</span><span class="op">(</span><span class="strut"> </span></p>
|
|
<p id="t96" class="pln"> <span class="nam">seq_len</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t97" class="pln"> <span class="nam">normalise</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'normalise'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t98" class="pln"> <span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t99" class="pln"><span class="strut"> </span></p>
|
|
<p id="t100" class="stm mis"> <span class="nam">predictions</span> <span class="op">=</span> <span class="nam">model</span><span class="op">.</span><span class="nam">predict_sequences_multiple</span><span class="op">(</span><span class="nam">x_test</span><span class="op">,</span> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t101" class="pln"> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t102" class="stm mis"> <span class="nam">print</span><span class="op">(</span><span class="str">"训练：\n"</span><span class="op">,</span> <span class="nam">predictions</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t103" class="pln"> <span class="com"># plot_results_multiple(predictions, y_test, configs['data']['sequence_length'])</span><span class="strut"> </span></p>
|
|
<p id="t104" class="pln"><span class="strut"> </span></p>
|
|
<p id="t105" class="pln"><span class="strut"> </span></p>
|
|
<p id="t106" class="pln"><span class="com"># 对指定公司的股票进行预测</span><span class="strut"> </span></p>
|
|
<p id="t107" class="stm run hide_run"><span class="key">def</span> <span class="nam">prediction</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">,</span> <span class="nam">real</span><span class="op">=</span><span class="key">True</span><span class="op">,</span> <span class="nam">pre_len</span><span class="op">=</span><span class="num">30</span><span class="op">,</span> <span class="nam">plot</span><span class="op">=</span><span class="key">False</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t108" class="pln"> <span class="str">'''</span><span class="strut"> </span></p>
|
|
<p id="t109" class="pln"><span class="str"> 使用保存的模型，对输入数据进行预测</span><span class="strut"> </span></p>
|
|
<p id="t110" class="pln"><span class="str"> '''</span><span class="strut"> </span></p>
|
|
<p id="t111" class="stm run hide_run"> <span class="nam">config_path</span> <span class="op">=</span> <span class="nam">get_config_path</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t112" class="stm run hide_run"> <span class="nam">configs</span> <span class="op">=</span> <span class="nam">json</span><span class="op">.</span><span class="nam">load</span><span class="op">(</span><span class="nam">open</span><span class="op">(</span><span class="nam">config_path</span><span class="op">,</span> <span class="str">'r'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t113" class="stm run hide_run"> <span class="nam">data</span> <span class="op">=</span> <span class="nam">DataLoader</span><span class="op">(</span><span class="strut"> </span></p>
|
|
<p id="t114" class="pln"> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">get_data_path</span><span class="op">(</span><span class="op">)</span><span class="op">,</span> <span class="nam">stock_code</span> <span class="op">+</span> <span class="str">".csv"</span><span class="op">)</span><span class="op">,</span> <span class="com"># configs['data']['filename']</span><span class="strut"> </span></p>
|
|
<p id="t115" class="pln"> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'train_test_split'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t116" class="pln"> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'columns'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t117" class="pln"> <span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t118" class="pln"><span class="strut"> </span></p>
|
|
<p id="t119" class="stm run hide_run"> <span class="nam">file_path</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="op">,</span><span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="str">"saved_models"</span><span class="op">,</span><span class="nam">stock_code</span> <span class="op">+</span> <span class="str">".h5"</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t120" class="stm run hide_run"> <span class="nam">model</span> <span class="op">=</span> <span class="nam">Model</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t121" class="stm run hide_run"> <span class="nam">keras</span><span class="op">.</span><span class="nam">backend</span><span class="op">.</span><span class="nam">clear_session</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t122" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">load_model</span><span class="op">(</span><span class="nam">file_path</span><span class="op">)</span> <span class="com"># 根据配置文件新建模型</span><span class="strut"> </span></p>
|
|
<p id="t123" class="pln"><span class="strut"> </span></p>
|
|
<p id="t124" class="pln"> <span class="com"># predict_length = configs['data']['sequence_length'] # 预测长度</span><span class="strut"> </span></p>
|
|
<p id="t125" class="stm run hide_run"> <span class="nam">predict_length</span> <span class="op">=</span> <span class="nam">pre_len</span><span class="strut"> </span></p>
|
|
<p id="t126" class="stm run hide_run"> <span class="key">if</span> <span class="nam">real</span><span class="op">:</span> <span class="com"># 用最近一个窗口的数据进行预测，没有对比数据</span><span class="strut"> </span></p>
|
|
<p id="t127" class="stm run hide_run"> <span class="nam">win_position</span> <span class="op">=</span> <span class="op">-</span><span class="num">1</span><span class="strut"> </span></p>
|
|
<p id="t128" class="pln"> <span class="key">else</span><span class="op">:</span> <span class="com"># 用指定位置的一个窗口数据进行预测，有对比真实数据（用于绘图对比）</span><span class="strut"> </span></p>
|
|
<p id="t129" class="stm mis"> <span class="nam">win_position</span> <span class="op">=</span> <span class="op">-</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t130" class="pln"><span class="strut"> </span></p>
|
|
<p id="t131" class="stm run hide_run"> <span class="nam">x_test</span><span class="op">,</span> <span class="nam">y_test</span> <span class="op">=</span> <span class="nam">data</span><span class="op">.</span><span class="nam">get_test_data</span><span class="op">(</span><span class="strut"> </span></p>
|
|
<p id="t132" class="pln"> <span class="nam">seq_len</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t133" class="pln"> <span class="nam">normalise</span><span class="op">=</span><span class="key">False</span><span class="strut"> </span></p>
|
|
<p id="t134" class="pln"> <span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t135" class="pln"><span class="strut"> </span></p>
|
|
<p id="t136" class="stm run hide_run"> <span class="nam">x_test</span> <span class="op">=</span> <span class="nam">x_test</span><span class="op">[</span><span class="nam">win_position</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t137" class="stm run hide_run"> <span class="nam">x_test</span> <span class="op">=</span> <span class="nam">x_test</span><span class="op">[</span><span class="nam">np</span><span class="op">.</span><span class="nam">newaxis</span><span class="op">,</span> <span class="op">:</span><span class="op">,</span> <span class="op">:</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t138" class="stm run hide_run"> <span class="key">if</span> <span class="key">not</span> <span class="nam">real</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t139" class="stm mis"> <span class="nam">y_test_real</span> <span class="op">=</span> <span class="nam">y_test</span><span class="op">[</span><span class="nam">win_position</span><span class="op">:</span><span class="nam">win_position</span> <span class="op">+</span> <span class="nam">predict_length</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t140" class="pln"><span class="strut"> </span></p>
|
|
<p id="t141" class="stm run hide_run"> <span class="nam">base</span> <span class="op">=</span> <span class="nam">x_test</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t142" class="stm run hide_run"> <span class="nam">print</span><span class="op">(</span><span class="str">"base value:\n"</span><span class="op">,</span> <span class="nam">base</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t143" class="pln"><span class="strut"> </span></p>
|
|
<p id="t144" class="stm run hide_run"> <span class="nam">x_test</span><span class="op">,</span> <span class="nam">y_test</span> <span class="op">=</span> <span class="nam">data</span><span class="op">.</span><span class="nam">get_test_data</span><span class="op">(</span><span class="strut"> </span></p>
|
|
<p id="t145" class="pln"> <span class="nam">seq_len</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t146" class="pln"> <span class="nam">normalise</span><span class="op">=</span><span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'normalise'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t147" class="pln"> <span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t148" class="stm run hide_run"> <span class="nam">x_test</span> <span class="op">=</span> <span class="nam">x_test</span><span class="op">[</span><span class="nam">win_position</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t149" class="stm run hide_run"> <span class="nam">x_test</span> <span class="op">=</span> <span class="nam">x_test</span><span class="op">[</span><span class="nam">np</span><span class="op">.</span><span class="nam">newaxis</span><span class="op">,</span> <span class="op">:</span><span class="op">,</span> <span class="op">:</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t150" class="pln"><span class="strut"> </span></p>
|
|
<p id="t151" class="pln"> <span class="com"># predictions = model.predict_sequences_multiple(x_test, configs['data']['sequence_length'],</span><span class="strut"> </span></p>
|
|
<p id="t152" class="pln"> <span class="com"># predict_length)</span><span class="strut"> </span></p>
|
|
<p id="t153" class="pln"><span class="strut"> </span></p>
|
|
<p id="t154" class="stm run hide_run"> <span class="nam">predictions</span> <span class="op">=</span> <span class="nam">model</span><span class="op">.</span><span class="nam">predict_1_win_sequence</span><span class="op">(</span><span class="nam">x_test</span><span class="op">,</span> <span class="nam">configs</span><span class="op">[</span><span class="str">'data'</span><span class="op">]</span><span class="op">[</span><span class="str">'sequence_length'</span><span class="op">]</span><span class="op">,</span> <span class="nam">predict_length</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t155" class="pln"> <span class="com"># 反归一化</span><span class="strut"> </span></p>
|
|
<p id="t156" class="stm run hide_run"> <span class="nam">predictions_array</span> <span class="op">=</span> <span class="nam">np</span><span class="op">.</span><span class="nam">array</span><span class="op">(</span><span class="nam">predictions</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t157" class="stm run hide_run"> <span class="nam">predictions_array</span> <span class="op">=</span> <span class="nam">base</span> <span class="op">*</span> <span class="op">(</span><span class="num">1</span> <span class="op">+</span> <span class="nam">predictions_array</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t158" class="stm run hide_run"> <span class="nam">predictions</span> <span class="op">=</span> <span class="nam">predictions_array</span><span class="op">.</span><span class="nam">tolist</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t159" class="pln"><span class="strut"> </span></p>
|
|
<p id="t160" class="pln"> <span class="com"># print("预测数据:\n", predictions)</span><span class="strut"> </span></p>
|
|
<p id="t161" class="stm run hide_run"> <span class="key">if</span> <span class="key">not</span> <span class="nam">real</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t162" class="stm mis"> <span class="nam">print</span><span class="op">(</span><span class="str">"真实数据：\n"</span><span class="op">,</span> <span class="nam">y_test_real</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t163" class="pln"><span class="strut"> </span></p>
|
|
<p id="t164" class="pln"> <span class="com"># plot_results_multiple(predictions, y_test, predict_length)</span><span class="strut"> </span></p>
|
|
<p id="t165" class="stm run hide_run"> <span class="key">if</span> <span class="nam">plot</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t166" class="stm mis"> <span class="key">if</span> <span class="nam">real</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t167" class="stm mis"> <span class="nam">plot_results</span><span class="op">(</span><span class="nam">predictions</span><span class="op">,</span> <span class="op">[</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t168" class="pln"> <span class="key">else</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t169" class="stm mis"> <span class="nam">plot_results</span><span class="op">(</span><span class="nam">predictions</span><span class="op">,</span> <span class="nam">y_test_real</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t170" class="pln"><span class="strut"> </span></p>
|
|
<p id="t171" class="stm run hide_run"> <span class="key">return</span> <span class="nam">format_predictions</span><span class="op">(</span><span class="nam">predictions</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t172" class="pln"><span class="strut"> </span></p>
|
|
<p id="t173" class="stm run hide_run"><span class="key">def</span> <span class="nam">format_predictions</span><span class="op">(</span><span class="nam">predictions</span><span class="op">)</span><span class="op">:</span> <span class="com"># 给预测数据添加对应日期</span><span class="strut"> </span></p>
|
|
<p id="t174" class="stm run hide_run"> <span class="nam">date_predict</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t175" class="stm run hide_run"> <span class="nam">cur</span> <span class="op">=</span> <span class="nam">datetime</span><span class="op">.</span><span class="nam">now</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t176" class="stm run hide_run"> <span class="nam">cur</span> <span class="op">+=</span> <span class="nam">timedelta</span><span class="op">(</span><span class="nam">days</span><span class="op">=</span><span class="num">1</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t177" class="stm run hide_run"> <span class="nam">counter</span> <span class="op">=</span> <span class="num">0</span><span class="strut"> </span></p>
|
|
<p id="t178" class="pln"><span class="strut"> </span></p>
|
|
<p id="t179" class="stm run hide_run"> <span class="key">while</span> <span class="nam">counter</span> <span class="op"><</span> <span class="nam">len</span><span class="op">(</span><span class="nam">predictions</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t180" class="stm run hide_run"> <span class="key">if</span> <span class="nam">cur</span><span class="op">.</span><span class="nam">isoweekday</span><span class="op">(</span><span class="op">)</span> <span class="op">==</span> <span class="num">6</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t181" class="stm run hide_run"> <span class="nam">cur</span> <span class="op">=</span> <span class="nam">cur</span> <span class="op">+</span> <span class="nam">timedelta</span><span class="op">(</span><span class="nam">days</span><span class="op">=</span><span class="num">2</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t182" class="stm run hide_run"> <span class="key">if</span> <span class="nam">cur</span><span class="op">.</span><span class="nam">isoweekday</span><span class="op">(</span><span class="op">)</span> <span class="op">==</span> <span class="num">7</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t183" class="stm run hide_run"> <span class="nam">cur</span> <span class="op">=</span> <span class="nam">cur</span> <span class="op">+</span> <span class="nam">timedelta</span><span class="op">(</span><span class="nam">days</span><span class="op">=</span><span class="num">1</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t184" class="stm run hide_run"> <span class="nam">date_predict</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="op">[</span><span class="nam">cur</span><span class="op">.</span><span class="nam">strftime</span><span class="op">(</span><span class="str">"%Y-%m-%d"</span><span class="op">)</span><span class="op">,</span><span class="nam">predictions</span><span class="op">[</span><span class="nam">counter</span><span class="op">]</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t185" class="stm run hide_run"> <span class="nam">cur</span> <span class="op">=</span> <span class="nam">cur</span> <span class="op">+</span> <span class="nam">timedelta</span><span class="op">(</span><span class="nam">days</span><span class="op">=</span><span class="num">1</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t186" class="stm run hide_run"> <span class="nam">counter</span> <span class="op">+=</span> <span class="num">1</span><span class="strut"> </span></p>
|
|
<p id="t187" class="pln"><span class="strut"> </span></p>
|
|
<p id="t188" class="stm run hide_run"> <span class="key">return</span> <span class="nam">date_predict</span><span class="strut"> </span></p>
|
|
<p id="t189" class="pln"><span class="strut"> </span></p>
|
|
<p id="t190" class="pln"><span class="str">'''</span><span class="strut"> </span></p>
|
|
<p id="t191" class="pln"><span class="str">def main(stock_code, train=False, predict=False):</span><span class="strut"> </span></p>
|
|
<p id="t192" class="pln"><span class="str"> configs = json.load(open(get_config_path(), 'r'))</span><span class="strut"> </span></p>
|
|
<p id="t193" class="pln"><span class="str"> companies = configs['companies']</span><span class="strut"> </span></p>
|
|
<p id="t194" class="pln"><span class="strut"> </span></p>
|
|
<p id="t195" class="pln"><span class="str"> if stock_code not in companies.keys():</span><span class="strut"> </span></p>
|
|
<p id="t196" class="pln"><span class="str"> print("该公司不在指定范围内")</span><span class="strut"> </span></p>
|
|
<p id="t197" class="pln"><span class="str"> return -1</span><span class="strut"> </span></p>
|
|
<p id="t198" class="pln"><span class="strut"> </span></p>
|
|
<p id="t199" class="pln"><span class="str"> if train:</span><span class="strut"> </span></p>
|
|
<p id="t200" class="pln"><span class="str"> train_model(stock_code)</span><span class="strut"> </span></p>
|
|
<p id="t201" class="pln"><span class="str"> return 0 # 训练完成</span><span class="strut"> </span></p>
|
|
<p id="t202" class="pln"><span class="strut"> </span></p>
|
|
<p id="t203" class="pln"><span class="str"> if predict:</span><span class="strut"> </span></p>
|
|
<p id="t204" class="pln"><span class="str"> # for root, dirs, files in os.walk('saved_models'):</span><span class="strut"> </span></p>
|
|
<p id="t205" class="pln"><span class="str"> # root:当前目录路径 dirs: 当前路径下所有子目录 files:当前路径下所有非目录子文件</span><span class="strut"> </span></p>
|
|
<p id="t206" class="pln"><span class="str"> if stock_code + ".h5" in os.listdir("saved_models"): # os.listdir:获得当前目录下的所有文件名。不包括子目录</span><span class="strut"> </span></p>
|
|
<p id="t207" class="pln"><span class="str"> return prediction(stock_code=stock_code, real=True, pre_len=20)</span><span class="strut"> </span></p>
|
|
<p id="t208" class="pln"><span class="str"> else:</span><span class="strut"> </span></p>
|
|
<p id="t209" class="pln"><span class="str"> return -2 # 该公司还没有训练模型</span><span class="strut"> </span></p>
|
|
<p id="t210" class="pln"><span class="str">'''</span><span class="strut"> </span></p>
|
|
<p id="t211" class="pln"> <span class="com"># 二维数组：[[data,value],[...]]</span><span class="strut"> </span></p>
|
|
<p id="t212" class="stm run hide_run"><span class="key">def</span> <span class="nam">get_hist_data</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">,</span> <span class="nam">recent_day</span><span class="op">=</span><span class="num">30</span><span class="op">)</span><span class="op">:</span> <span class="com"># 获取某股票，指定天数的历史close数据,包含日期</span><span class="strut"> </span></p>
|
|
<p id="t213" class="stm run hide_run"> <span class="nam">get_single_last_data</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t214" class="stm run hide_run"> <span class="nam">root_dir</span> <span class="op">=</span> <span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t215" class="stm run hide_run"> <span class="nam">file_path</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">root_dir</span><span class="op">,</span> <span class="str">"data/"</span> <span class="op">+</span> <span class="nam">stock_code</span> <span class="op">+</span> <span class="str">".csv"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t216" class="stm run hide_run"> <span class="nam">cols</span> <span class="op">=</span> <span class="op">[</span><span class="str">'Date'</span><span class="op">,</span> <span class="str">'Close'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t217" class="stm run hide_run"> <span class="nam">data_frame</span> <span class="op">=</span> <span class="nam">pd</span><span class="op">.</span><span class="nam">read_csv</span><span class="op">(</span><span class="nam">file_path</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t218" class="stm run hide_run"> <span class="nam">close_data</span> <span class="op">=</span> <span class="nam">data_frame</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="nam">cols</span><span class="op">)</span><span class="op">.</span><span class="nam">values</span><span class="op">[</span><span class="op">-</span><span class="nam">recent_day</span><span class="op">:</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t219" class="stm run hide_run"> <span class="key">return</span> <span class="nam">close_data</span><span class="op">.</span><span class="nam">tolist</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t220" class="pln"><span class="strut"> </span></p>
|
|
<p id="t221" class="pln"><span class="strut"> </span></p>
|
|
<p id="t222" class="stm run hide_run"><span class="key">def</span> <span class="nam">train_all_stock</span><span class="op">(</span><span class="op">)</span><span class="op">:</span> <span class="com">#</span><span class="strut"> </span></p>
|
|
<p id="t223" class="stm run hide_run"> <span class="nam">get_all_last_data</span><span class="op">(</span><span class="nam">start_date</span><span class="op">=</span><span class="str">"2010-01-01"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t224" class="stm run hide_run"> <span class="nam">configs</span> <span class="op">=</span> <span class="nam">json</span><span class="op">.</span><span class="nam">load</span><span class="op">(</span><span class="nam">open</span><span class="op">(</span><span class="nam">get_config_path</span><span class="op">(</span><span class="op">)</span><span class="op">,</span> <span class="str">'r'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t225" class="stm run hide_run"> <span class="nam">companies</span> <span class="op">=</span> <span class="nam">configs</span><span class="op">[</span><span class="str">'companies'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t226" class="stm run hide_run"> <span class="key">for</span> <span class="nam">stock_code</span> <span class="key">in</span> <span class="nam">companies</span><span class="op">.</span><span class="nam">keys</span><span class="op">(</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t227" class="stm run hide_run"> <span class="nam">train_model</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t228" class="pln"><span class="strut"> </span></p>
|
|
<p id="t229" class="stm run hide_run"> <span class="key">return</span> <span class="num">0</span><span class="strut"> </span></p>
|
|
<p id="t230" class="pln"><span class="strut"> </span></p>
|
|
<p id="t231" class="pln"><span class="strut"> </span></p>
|
|
<p id="t232" class="stm run hide_run"><span class="key">def</span> <span class="nam">predict_all_stock</span><span class="op">(</span><span class="nam">pre_len</span><span class="op">=</span><span class="num">10</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t233" class="stm run hide_run"> <span class="nam">file_path</span> <span class="op">=</span> <span class="nam">get_config_path</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t234" class="stm run hide_run"> <span class="nam">configs</span> <span class="op">=</span> <span class="nam">json</span><span class="op">.</span><span class="nam">load</span><span class="op">(</span><span class="nam">open</span><span class="op">(</span><span class="nam">file_path</span><span class="op">,</span> <span class="str">'r'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t235" class="stm run hide_run"> <span class="nam">companies</span> <span class="op">=</span> <span class="nam">configs</span><span class="op">[</span><span class="str">'companies'</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t236" class="stm run hide_run"> <span class="nam">predict_list</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t237" class="stm run hide_run"> <span class="key">for</span> <span class="nam">stock_code</span> <span class="key">in</span> <span class="nam">companies</span><span class="op">.</span><span class="nam">keys</span><span class="op">(</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t238" class="stm run hide_run"> <span class="nam">predict_list</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">prediction</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">=</span><span class="nam">stock_code</span><span class="op">,</span> <span class="nam">real</span><span class="op">=</span><span class="key">True</span><span class="op">,</span> <span class="nam">pre_len</span><span class="op">=</span><span class="nam">pre_len</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t239" class="pln"><span class="strut"> </span></p>
|
|
<p id="t240" class="stm run hide_run"> <span class="key">return</span> <span class="nam">predict_list</span><span class="strut"> </span></p>
|
|
<p id="t241" class="pln"><span class="strut"> </span></p>
|
|
<p id="t242" class="pln"><span class="strut"> </span></p>
|
|
<p id="t243" class="stm run hide_run"><span class="key">def</span> <span class="nam">get_config_path</span><span class="op">(</span><span class="op">)</span><span class="op">:</span> <span class="com"># config.json的绝对路径</span><span class="strut"> </span></p>
|
|
<p id="t244" class="stm run hide_run"> <span class="nam">root_dir</span> <span class="op">=</span> <span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t245" class="stm run hide_run"> <span class="key">return</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">root_dir</span><span class="op">,</span> <span class="str">"config.json"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t246" class="pln"><span class="strut"> </span></p>
|
|
<p id="t247" class="pln"><span class="strut"> </span></p>
|
|
<p id="t248" class="stm run hide_run"><span class="key">def</span> <span class="nam">get_data_path</span><span class="op">(</span><span class="op">)</span><span class="op">:</span> <span class="com"># data目录的绝对路径</span><span class="strut"> </span></p>
|
|
<p id="t249" class="stm run hide_run"> <span class="nam">root_dir</span> <span class="op">=</span> <span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t250" class="stm run hide_run"> <span class="key">return</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">root_dir</span><span class="op">,</span> <span class="str">"data"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t251" class="pln"><span class="strut"> </span></p>
|
|
<p id="t252" class="pln"><span class="strut"> </span></p>
|
|
<p id="t253" class="stm run hide_run"><span class="key">def</span> <span class="nam">get_parent_dir</span><span class="op">(</span><span class="op">)</span><span class="op">:</span> <span class="com"># 当前文件的父目录绝对路径</span><span class="strut"> </span></p>
|
|
<p id="t254" class="stm run hide_run"> <span class="key">return</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">dirname</span><span class="op">(</span><span class="nam">__file__</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t255" class="pln"><span class="strut"> </span></p>
|
|
<p id="t256" class="pln"><span class="strut"> </span></p>
|
|
<p id="t257" class="pln"><span class="com"># if __name__ == '__main__':</span><span class="strut"> </span></p>
|
|
<p id="t258" class="pln"><span class="com"># # get_all_last_data("2010-01-01") # 先获得最新数据</span><span class="strut"> </span></p>
|
|
<p id="t259" class="pln"><span class="com"># train_all_stock()</span><span class="strut"> </span></p>
|
|
<p id="t260" class="pln"><span class="com"># # predict_all_stock()</span><span class="strut"> </span></p>
|
|
|
|
</td>
|
|
</tr>
|
|
</table>
|
|
</div>
|
|
|
|
<div id="footer">
|
|
<div class="content">
|
|
<p>
|
|
<a class="nav" href="index.html">« index</a> <a class="nav" href="https://coverage.readthedocs.io">coverage.py v4.5.2</a>,
|
|
created at 2018-12-22 16:41
|
|
</p>
|
|
</div>
|
|
</div>
|
|
|
|
</body>
|
|
</html>
|