Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
Menu
Open sidebar
ITMO-NSS-team
FEDOT
Commits
617cc8dc
Commit
617cc8dc
authored
3 years ago
by
valer1435
Browse files
Options
Download
Email Patches
Plain Diff
bug with test_cv_api_correct
parent
9fd0447a
bug_test_cv_api_correct
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
test/unit/validation/test_table_cv.py
+19
-0
test/unit/validation/test_table_cv.py
with
19 additions
and
0 deletions
+19
-0
test/unit/validation/test_table_cv.py
View file @
617cc8dc
...
...
@@ -150,3 +150,22 @@ def test_cv_api_correct():
assert
isinstance
(
fedot_model
,
Pipeline
)
assert
len
(
prediction
)
==
len
(
dataset_to_validate
.
target
)
assert
metric
[
'f1'
]
>
0
# TODO exception here
def
test_resample_pipeline
():
resample
=
PrimaryNode
(
"resample"
)
resample
.
custom_params
=
{
"balance"
:
"expand_minority"
,
"replace"
:
True
,
"n_samples"
:
0.5
}
scaling
=
SecondaryNode
(
"scaling"
,
nodes_from
=
[
resample
])
estimator
=
SecondaryNode
(
"xgboost"
,
nodes_from
=
[
scaling
])
model
=
Pipeline
(
estimator
)
task
=
Task
(
task_type
=
TaskTypesEnum
.
classification
)
dataset_to_compose
,
dataset_to_validate
=
get_data
(
task
)
fedot_model
=
model
.
fit
(
dataset_to_compose
)
prediction
=
model
.
predict
(
dataset_to_validate
)
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment
Menu
Projects
Groups
Snippets
Help