ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-09 02:48:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Make xy_sample receive features, not instances

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-03-31 09:57:57 -05:00
parent 8fc9979b37
commit fe7bad885c
12 changed files with 158 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/components/test_component.py
View File

@@ -7,8 +7,7 @@ from miplearn import Component, Instance
def test_xy_instance():
def _xy_sample(instance, sample):
print(sample)
def _xy_sample(features, sample):
x = {
"s1": {
"category_a": [
@@ -54,8 +53,10 @@ def test_xy_instance():
comp = Component()
instance_1 = Mock(spec=Instance)
instance_1.training_data = ["s1", "s2"]
instance_1.features = {}
instance_2 = Mock(spec=Instance)
instance_2.training_data = ["s3"]
instance_2.features = {}
comp.xy_sample = _xy_sample
x_expected = {
"category_a": [