pe</span><span class="o">=</span><span class="nb">object</span><span class="p">)</span>
    <span class="n">train_idx</span><span class="p">,</span> <span class="n">_</span> <span class="o">=</span> <span class="n">train_test_split</span><span class="p">(</span>
        <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">adata</span><span class="p">)),</span> <span class="n">test_size</span><span class="o">=</span><span class="mf">0.5</span