jebus/enviPy-bayer
1
0
Fork 0
forked from enviPath/enviPy
Code Issues Pull Requests Actions Releases Activity

fixed calls to create MLRelativeReasoning during bootstrap.py and test_model.py

Browse Source
This commit is contained in:
Liam Brydon
2025-08-22 11:28:01 +12:00
parent b95ec98a2f
commit b45c99f7d3
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
epdb/management/commands/bootstrap.py
View File

@ -187,17 +187,17 @@ class Command(BaseCommand):
# Create RR # Create RR
ml_model = MLRelativeReasoning.create( ml_model = MLRelativeReasoning.create(
pack, package=pack,
'ECC - BBD - T0.5', rule_packages=[mapping['EAWAG-BBD']],
'ML Relative Reasoning', data_packages=[mapping['EAWAG-BBD']],
[mapping['EAWAG-BBD']], eval_packages=[],
[mapping['EAWAG-BBD']], threshold=0.5,
[], name='ECC - BBD - T0.5',
0.5 description='ML Relative Reasoning',
) )
X, y = ml_model.build_dataset() ml_model.build_dataset()
ml_model.build_model(X, y) ml_model.build_model()
# ml_model.evaluate_model() # ml_model.evaluate_model()
# If available create EnviFormerModel # If available create EnviFormerModel

6
tests/test_model.py
View File

@ -37,12 +37,12 @@ class ModelTest(TestCase):
mod = MLRelativeReasoning.create( mod = MLRelativeReasoning.create(
self.package, self.package,
'ECC - BBD - 0.5',
'Created MLRelativeReasoning in Testcase',
rule_package_objs, rule_package_objs,
data_package_objs, data_package_objs,
eval_packages_objs, eval_packages_objs,
threshold threshold,
'ECC - BBD - 0.5',
'Created MLRelativeReasoning in Testcase',
) )
ds = mod.load_dataset() ds = mod.load_dataset()