Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add more parameters to the MLP training method #242

Open
3 tasks
FlorentinD opened this issue Jan 18, 2023 · 0 comments
Open
3 tasks

Add more parameters to the MLP training method #242

FlorentinD opened this issue Jan 18, 2023 · 0 comments
Labels
feature request A suggestion for a new feature good first issue Indicates a good issue for first-time contributors

Comments

@FlorentinD
Copy link
Contributor

Is your feature request related to a problem? Please describe.
In the machine learning pipelines, you can use different trainer methods when adding a model candidate, such as RandomForest or MLP.
For MLP, we could expose more parameters to configure.

Current exposed parameters are listed at https://neo4j.com/docs/graph-data-science/current/machine-learning/training-methods/mlp/.

Describe the solution you would like

We could also expose other optional parameters such as:

  • Different activiation functions (in addition to Softmax)
  • Dropouts
  • Optimizer

These are just examples, you could also come up with others.

Additional context

Good starting points are MLPClassifierTrainConfig and MLPClassifier

If you want to work on this issue please drop a comment :)

@FlorentinD FlorentinD added feature request A suggestion for a new feature good first issue Indicates a good issue for first-time contributors labels Jan 18, 2023
@FlorentinD FlorentinD changed the title Add more parameters to MLP trainer method Add more parameters to the MLP trainer method Jan 18, 2023
@FlorentinD FlorentinD changed the title Add more parameters to the MLP trainer method Add more parameters to the MLP training method Jan 18, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request A suggestion for a new feature good first issue Indicates a good issue for first-time contributors
Projects
None yet
Development

No branches or pull requests

1 participant