Skip to content

Update pipelines.py#157

Open
Mikehade wants to merge 2 commits into
squaredev-io:mainfrom
Mikehade:patch-1
Open

Update pipelines.py#157
Mikehade wants to merge 2 commits into
squaredev-io:mainfrom
Mikehade:patch-1

Conversation

@Mikehade

@Mikehade Mikehade commented Mar 15, 2023

Copy link
Copy Markdown

Update pipelines.py by adding stratify parameter to train_test_split function of the sklearn library

  • closes Adjust the train_test_split parts in pipelines #91 (Replace 91 with the GitHub issue number)
  • Tests added and passed if fixing a bug or adding a new feature.
  • All code checks passed.
  • Added type annotations to new arguments/methods/functions.
  • Updated documentation.

Update pipelines.py by adding stratify parameter to train_test_split function of the sklearn library
@Mikehade

Copy link
Copy Markdown
Author

Stratify parameter added added

@momegas momegas left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is great

@momegas

momegas commented Mar 20, 2023

Copy link
Copy Markdown
Member

Some tests are failing 🥶

@Mikehade

Copy link
Copy Markdown
Author

I will check out the errors to find a solution

@Mikehade

Copy link
Copy Markdown
Author

I have corrected the stratify variable

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Adjust the train_test_split parts in pipelines

2 participants