dotnet/machinelearning
Voir sur GitHubAdd FieldAwareFactorizationMachine to AutoML
Open
#3 985 ouverte le 10 juil. 2019
AutoML.NETP2command-linegood first issue
Métriques du dépôt
- Stars
- (9 125 stars)
- Métriques de merge PR
- (Merge moyen 6j 19h) (5 PRs mergées en 30 j)
Description
FieldAwareFactorizationMachine is good for large dataset like the Criteo 1TB dataset.
Currently FieldAwareFactorizationMachine is not swept over in AutoML.
Task:
- Add trainer to default list of binary learners to try
- Add sweep range
- Add to CLI's C# CodeGen
Should be easy to just replicate an existing trainer like SDCA: https://github.com/dotnet/machinelearning/blob/d518b587b06ac3896a48646622b0f2169a230855/src/Microsoft.ML.AutoML/TrainerExtensions/BinaryTrainerExtensions.cs#L150-L169