Skip to content

Commit

Permalink
deploy: 72e9508
Browse files Browse the repository at this point in the history
  • Loading branch information
@Demirrr
Demirrr committed Oct 23, 2024
1 parent 9eb1521 commit 096cf3f
Show file tree
Hide file tree
Showing 4 changed files with 2 additions and 5 deletions.
2 changes: 1 addition & 1 deletion _modules/dicee/executer.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ <h1>Source code for dicee.executer</h1><div class="highlight"><pre>
<span class="c1"># Avoid</span>
<span class="kn">from</span> <span class="nn">.static_preprocess_funcs</span> <span class="kn">import</span> <span class="n">preprocesses_input_args</span>
<span class="kn">from</span> <span class="nn">.trainer</span> <span class="kn">import</span> <span class="n">DICE_Trainer</span>
<span class="kn">from</span> <span class="nn">.static_funcs</span> <span class="kn">import</span> <span class="n">timeit</span><span class="p">,</span> <span class="n">continual_training_setup_executor</span><span class="p">,</span> <span class="n">read_or_load_kg</span><span class="p">,</span> <span class="n">load_json</span><span class="p">,</span> <span class="n">store</span><span class="p">,</span> <span class="n">create_experiment_folder</span>
<span class="kn">from</span> <span class="nn">.static_funcs</span> <span class="kn">import</span> <span class="n">timeit</span><span class="p">,</span> <span class="n">read_or_load_kg</span><span class="p">,</span> <span class="n">load_json</span><span class="p">,</span> <span class="n">store</span><span class="p">,</span> <span class="n">create_experiment_folder</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>

<span class="n">logging</span><span class="o">.</span><span class="n">getLogger</span><span class="p">(</span><span class="s1">&#39;pytorch_lightning&#39;</span><span class="p">)</span><span class="o">.</span><span class="n">setLevel</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span>
Expand Down
2 changes: 0 additions & 2 deletions _modules/dicee/knowledge_graph.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -93,8 +93,6 @@ <h1>Source code for dicee.knowledge_graph</h1><div class="highlight"><pre>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">pandas</span> <span class="k">as</span> <span class="nn">pd</span>
<span class="kn">import</span> <span class="nn">polars</span> <span class="k">as</span> <span class="nn">pl</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">from</span> <span class="nn">.read_preprocess_save_load_kg.util</span> <span class="kn">import</span> <span class="n">load_numpy_ndarray</span>
<div class="viewcode-block" id="KG">
<a class="viewcode-back" href="../../autoapi/dicee/knowledge_graph/index.html#dicee.knowledge_graph.KG">[docs]</a>
<span class="k">class</span> <span class="nc">KG</span><span class="p">:</span>
Expand Down
3 changes: 1 addition & 2 deletions _modules/dicee/trainer/dice_trainer.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,6 @@
<h1>Source code for dicee.trainer.dice_trainer</h1><div class="highlight"><pre>
<span></span><span class="kn">import</span> <span class="nn">lightning</span> <span class="k">as</span> <span class="nn">pl</span>
<span class="kn">import</span> <span class="nn">polars</span>
<span class="kn">import</span> <span class="nn">gc</span>
<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Union</span>
<span class="kn">from</span> <span class="nn">dicee.models.base_model</span> <span class="kn">import</span> <span class="n">BaseKGE</span>
<span class="kn">from</span> <span class="nn">dicee.static_funcs</span> <span class="kn">import</span> <span class="n">select_model</span>
Expand Down Expand Up @@ -421,7 +420,7 @@ <h1>Source code for dicee.trainer.dice_trainer</h1><div class="highlight"><pre>
<span class="n">train_set_for_i_th_fold</span><span class="p">,</span> <span class="n">test_set_for_i_th_fold</span> <span class="o">=</span> <span class="n">dataset</span><span class="o">.</span><span class="n">train_set</span><span class="p">[</span><span class="n">train_index</span><span class="p">],</span> <span class="n">dataset</span><span class="o">.</span><span class="n">train_set</span><span class="p">[</span>
<span class="n">test_index</span><span class="p">]</span>

<span class="n">trainer</span><span class="o">.</span><span class="n">fit</span><span class="p">(</span><span class="n">model</span><span class="p">,</span> <span class="n">train_dataloaders</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">initialize_dataloader</span><span class="p">(</span>
<span class="n">trainer</span><span class="o">.</span><span class="n">fit</span><span class="p">(</span><span class="n">model</span><span class="p">,</span> <span class="n">train_dataloaders</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">init_dataloader</span><span class="p">(</span>
<span class="n">construct_dataset</span><span class="p">(</span><span class="n">train_set</span><span class="o">=</span><span class="n">train_set_for_i_th_fold</span><span class="p">,</span>
<span class="n">entity_to_idx</span><span class="o">=</span><span class="n">dataset</span><span class="o">.</span><span class="n">entity_to_idx</span><span class="p">,</span>
<span class="n">relation_to_idx</span><span class="o">=</span><span class="n">dataset</span><span class="o">.</span><span class="n">relation_to_idx</span><span class="p">,</span>
Expand Down
Binary file modified diceembeddings.pdf
View file
Binary file not shown.

0 comments on commit 096cf3f

Please sign in to comment.