Skip to content
Snippets Groups Projects
Commit a6476c0e authored by Florian Grosse's avatar Florian Grosse
Browse files

Changed split method for stratified split from adapter, fixes Issue #2

parent 07c2dd01
No related branches found
No related tags found
No related merge requests found
......@@ -9,6 +9,7 @@ from multiprocessing import Manager
from conML.shared.logger import ReconstructionInfo
from conML.domain.reconstruction.selection import WinnerSelector
from conML.ports import split
InterimPragmatic = namedtuple("InterimPragmatic", "model, predictions")
......@@ -560,16 +561,6 @@ class Reconstructor:
self.settings.min_reliability = value
def split(learnblock, reliability_sample):
indices = learnblock.indices()
eval_size = int(learnblock.n_rows() * reliability_sample)
eval_idx, train_idx = indices[:eval_size], indices[eval_size:]
return (
learnblock.new_block_from(train_idx, index=True),
learnblock.new_block_from(eval_idx, index=True)
)
def valid_reconstructed(model, knowledge_domain, settings):
"""Test if the trained supersived model fullfills the accuracy criteria.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment