Skip to content
This repository has been archived by the owner on May 27, 2022. It is now read-only.

Feature/6 kwargs #13

Merged
merged 6 commits into from
Feb 3, 2021
Merged

Feature/6 kwargs #13

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
d868725
Update sphinx
Huffon Feb 3, 2021
3c0b96b
Add kwrags for predict func #6
Huffon Feb 3, 2021
15ef2ba
Add test cases
Huffon Feb 3, 2021
3b9b7f1
Merge branch 'master' of https://github.com/kakaobrain/pororo into fe…
Huffon Feb 3, 2021
aad4f47
Fix device type error
Huffon Feb 3, 2021
f27ab5d
Update sphinx
Huffon Feb 3, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file modified docs/_build/doctrees/environment.pickle
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/asr.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/caption.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/collocation.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/inflection.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/lemma.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/ocr.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/tokenization.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/miscs/wt.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/seq2seq/const.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/seq2seq/g2p.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/seq2seq/gec.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/seq2seq/mt.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/seq2seq/p2g.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/seq2seq/para_gen.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/seq2seq/summary.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/tagging/contextualized.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/tagging/dp.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/tagging/fill.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/tagging/mrc.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/tagging/ner.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/tagging/pos.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/tagging/srl.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/aes.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/age_suitability.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/nli.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/para_id.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/review_score.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/sent2vec.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/sentiment.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/sts.doctree
View file
Open in desktop
Binary file not shown.
Binary file modified docs/_build/doctrees/text_cls/zero.doctree
View file
Open in desktop
Binary file not shown.
8 changes: 2 additions & 6 deletions docs/_build/html/_modules/pororo/tasks/age_suitability.html
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -367,7 +367,7 @@ <h1>Source code for pororo.tasks.age_suitability</h1><div class="highlight"><pre

<span class="k">return</span> <span class="n">result</span>

<div class="viewcode-block" id="PororoBertMovie.predict"><a class="viewcode-back" href="../../../text_cls/age_suitability.html#pororo.tasks.age_suitability.PororoBertMovie.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">script</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
<div class="viewcode-block" id="PororoBertMovie.predict"><a class="viewcode-back" href="../../../text_cls/age_suitability.html#pororo.tasks.age_suitability.PororoBertMovie.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">script</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Conduct (movie) script analysis</span>

Expand Down Expand Up @@ -399,11 +399,7 @@ <h1>Source code for pororo.tasks.age_suitability</h1><div class="highlight"><pre
<span class="s2">&quot;emotion&quot;</span><span class="p">:</span> <span class="n">emotion</span><span class="p">,</span>
<span class="p">}</span>

<span class="k">return</span> <span class="n">result</span></div>

<span class="k">def</span> <span class="fm">__call__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">text</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">show_probs</span><span class="o">=</span><span class="kc">False</span><span class="p">):</span>
<span class="k">assert</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">text</span><span class="p">,</span> <span class="nb">str</span><span class="p">),</span> <span class="s2">&quot;Input text should be string type&quot;</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">predict</span><span class="p">(</span><span class="n">text</span><span class="p">)</span></div>
<span class="k">return</span> <span class="n">result</span></div></div>


<div class="viewcode-block" id="RobertaEncoder"><a class="viewcode-back" href="../../../text_cls/age_suitability.html#pororo.tasks.age_suitability.RobertaEncoder">[docs]</a><span class="k">class</span> <span class="nc">RobertaEncoder</span><span class="p">(</span><span class="n">nn</span><span class="o">.</span><span class="n">Module</span><span class="p">):</span>
Expand Down
2 changes: 1 addition & 1 deletion docs/_build/html/_modules/pororo/tasks/automated_essay_scoring.html
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -263,7 +263,7 @@ <h1>Source code for pororo.tasks.automated_essay_scoring</h1><div class="highlig
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="n">config</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_model</span> <span class="o">=</span> <span class="n">model</span>

<div class="viewcode-block" id="PororoBertAes.predict"><a class="viewcode-back" href="../../../text_cls/aes.html#pororo.tasks.automated_essay_scoring.PororoBertAes.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sent</span><span class="p">:</span> <span class="nb">str</span><span class="p">):</span>
<div class="viewcode-block" id="PororoBertAes.predict"><a class="viewcode-back" href="../../../text_cls/aes.html#pororo.tasks.automated_essay_scoring.PororoBertAes.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sent</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Conduct Automated Essay Scoring</span>

Expand Down
23 changes: 5 additions & 18 deletions docs/_build/html/_modules/pororo/tasks/automatic_speech_recognition.html
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -391,9 +391,7 @@ <h1>Source code for pororo.tasks.automatic_speech_recognition</h1><div class="hi
<div class="viewcode-block" id="PororoASR.predict"><a class="viewcode-back" href="../../../miscs/asr.html#pororo.tasks.automatic_speech_recognition.PororoASR.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="n">audio_path</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">top_db</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">48</span><span class="p">,</span>
<span class="n">batch_size</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">1</span><span class="p">,</span>
<span class="n">vad</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Conduct speech recognition for audio in a given path</span>
Expand All @@ -409,6 +407,9 @@ <h1>Source code for pororo.tasks.automatic_speech_recognition</h1><div class="hi
<span class="sd"> dict: result of speech recognition</span>

<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">top_db</span> <span class="o">=</span> <span class="n">kwargs</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;top_db&quot;</span><span class="p">,</span> <span class="mi">48</span><span class="p">)</span>
<span class="n">batch_size</span> <span class="o">=</span> <span class="n">kwargs</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;batch_size&quot;</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span>
<span class="n">vad</span> <span class="o">=</span> <span class="n">kwargs</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;batch_size&quot;</span><span class="p">,</span> <span class="kc">False</span><span class="p">)</span>

<span class="n">signal</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_preprocess_audio</span><span class="p">(</span><span class="n">audio_path</span><span class="p">)</span>

Expand All @@ -418,21 +419,7 @@ <h1>Source code for pororo.tasks.automatic_speech_recognition</h1><div class="hi
<span class="n">top_db</span><span class="o">=</span><span class="n">top_db</span><span class="p">,</span>
<span class="n">vad</span><span class="o">=</span><span class="n">vad</span><span class="p">,</span>
<span class="n">batch_size</span><span class="o">=</span><span class="n">batch_size</span><span class="p">,</span>
<span class="p">)</span></div>

<span class="k">def</span> <span class="fm">__call__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="n">audio_path</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">top_db</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">48</span><span class="p">,</span>
<span class="n">batch_size</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">16</span><span class="p">,</span>
<span class="n">vad</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">predict</span><span class="p">(</span>
<span class="n">audio_path</span><span class="o">=</span><span class="n">audio_path</span><span class="p">,</span>
<span class="n">top_db</span><span class="o">=</span><span class="n">top_db</span><span class="p">,</span>
<span class="n">batch_size</span><span class="o">=</span><span class="n">batch_size</span><span class="p">,</span>
<span class="n">vad</span><span class="o">=</span><span class="n">vad</span><span class="p">,</span>
<span class="p">)</span></div>
<span class="p">)</span></div></div>
</pre></div>

</div>
Expand Down
2 changes: 1 addition & 1 deletion docs/_build/html/_modules/pororo/tasks/collocation.html
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -313,7 +313,7 @@ <h1>Source code for pororo.tasks.collocation</h1><div class="highlight"><pre>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="n">config</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_model</span> <span class="o">=</span> <span class="n">model</span>

<div class="viewcode-block" id="PororoCollocate.predict"><a class="viewcode-back" href="../../../miscs/collocation.html#pororo.tasks.collocation.PororoCollocate.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">text</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
<div class="viewcode-block" id="PororoCollocate.predict"><a class="viewcode-back" href="../../../miscs/collocation.html#pororo.tasks.collocation.PororoCollocate.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">text</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Conduct collocation search using index file</span>

Expand Down
Loading