Skip to content
This repository has been archived by the owner on Aug 26, 2024. It is now read-only.

Add documentation for what functions invoked in model.train as in model.predict #180

Merged
merged 3 commits into from
Oct 5, 2022

Conversation

bilal-aamer
Copy link
Contributor

@bilal-aamer bilal-aamer commented Oct 4, 2022

#148
Updated the docs of the unionml.model.Model.train method to show what functions are invoked as done in the model.predict method.

unionml/model.py Outdated
@@ -514,15 +514,19 @@ def train(

:param hyperparameters: a dictionary mapping hyperparameter names to values. This is passed into the
``init`` callable to initialize a model object.
:param loader_kwargs: key-word arguments to pass to the registered :meth:`unionml.Dataset.loader` function.
:param loader_kwargs: key-word arguments to pass to the registered
Copy link
Contributor

Choose a reason for hiding this comment

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

hi @bilal-aamer wanted to clarify the updates needed for #148.

It needs to list the tasks that are invoked in the training workflow.

Can you revert the changes that you made below and add a paragraph below the argument docstrings, something like:

The train method invokes an execution graph that composes together the following functions to train and evaluate a model:

- :meth:`unionml.Dataset.reader`
- :meth:`unionml.Dataset.loader`
- :meth:`unionml.Dataset.splitter`
- :meth:`unionml.Dataset.parser`
- :meth:`unionml.Model.trainer`
- :meth:`unionml.Model.predictor`
- :meth:`unionml.Model.evaluator`

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Sure will do

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Done! @cosmicBboy ! Please Review!

Copy link
Contributor

@cosmicBboy cosmicBboy left a comment

Choose a reason for hiding this comment

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

thanks @bilal-aamer !

@bilal-aamer
Copy link
Contributor Author

No, thank you @cosmicBboy 👍🏼

@cosmicBboy cosmicBboy merged commit 10d3d7e into unionai-oss:main Oct 5, 2022
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants