<a id="examples" class="tsd-anchor"></a><h2 class="tsd-anchor-link">Examples<a href="#examples" aria-label="Permalink" class="tsd-anchor-icon"><svg viewBox="0 0 24 24"><use href="assets/icons.svg#icon-anchor"></use></svg></a></h2><a id="ridge" class="tsd-anchor"></a><h3 class="tsd-anchor-link">Ridge<a href="#ridge" aria-label="Permalink" class="tsd-anchor-icon"><svg viewBox="0 0 24 24"><use href="assets/icons.svg#icon-anchor"></use></svg></a></h3><pre><code class="JavaScript"><span class="hl-9">import</span><span class="hl-1"> </span><span class="hl-10">dam</span><span class="hl-1"> </span><span class="hl-9">from</span><span class="hl-1"> </span><span class="hl-2">'@ai-on-browser/data-analysis-models'</span><span class="hl-1">;</span><br/><br/><span class="hl-3">const</span><span class="hl-1"> </span><span class="hl-12">x</span><span class="hl-1"> = </span><span class="hl-10">dam</span><span class="hl-1">.</span><span class="hl-10">Matrix</span><span class="hl-1">.</span><span class="hl-0">randn</span><span class="hl-1">(</span><span class="hl-13">100</span><span class="hl-1">, </span><span class="hl-13">3</span><span class="hl-1">);</span><br/><span class="hl-3">const</span><span class="hl-1"> </span><span class="hl-12">y</span><span class="hl-1"> = </span><span class="hl-10">x</span><span class="hl-1">.</span><span class="hl-0">sum</span><span class="hl-1">(</span><span class="hl-13">1</span><span class="hl-1">);</span><br/><br/><span class="hl-3">const</span><span class="hl-1"> </span><span class="hl-12">model</span><span class="hl-1"> = </span><span class="hl-3">new</span><span class="hl-1"> </span><span class="hl-10">dam</span><span class="hl-1">.</span><span class="hl-10">models</span><span class="hl-1">.</span><span class="hl-0">Ridge</span><span class="hl-1">(</span><span class="hl-13">0.1</span><span class="hl-1">);</span><br/><span class="hl-10">model</span><span class="hl-1">.</span><span class="hl-0">fit</span><span class="hl-1">(</span><span class="hl-10">x</span><span class="hl-1">.</span><span class="hl-0">toArray</span><span class="hl-1">(), </span><span class="hl-10">y</span><span class="hl-1">.</span><span class="hl-0">toArray</span><span class="hl-1">());</span><br/><br/><span class="hl-3">const</span><span class="hl-1"> </span><span class="hl-12">predict</span><span class="hl-1"> = </span><span class="hl-10">model</span><span class="hl-1">.</span><span class="hl-0">predict</span><span class="hl-1">(</span><span class="hl-10">x</span><span class="hl-1">.</span><span class="hl-0">toArray</span><span class="hl-1">());</span><br/><span class="hl-3">const</span><span class="hl-1"> </span><span class="hl-12">error</span><span class="hl-1"> = </span><span class="hl-10">dam</span><span class="hl-1">.</span><span class="hl-10">evaluate</span><span class="hl-1">.</span><span class="hl-0">rmse</span><span class="hl-1">(</span><span class="hl-10">predict</span><span class="hl-1">, </span><span class="hl-10">y</span><span class="hl-1">.</span><span class="hl-0">toArray</span><span class="hl-1">());</span><br/><span class="hl-10">console</span><span class="hl-1">.</span><span class="hl-0">log</span><span class="hl-1">(</span><span class="hl-10">error</span><span class="hl-1">);</span>
0 commit comments