dotnet/machinelearning
Add FieldAwareFactorizationMachine to AutoML
Ouverte
#3 985 ouverte le 10 juil. 2019
AutoML.NETP2command-linegood first issuein-pr
Métriques du dépôt
- Stars
- (9 125 étoiles)
- Métriques de merge PR
- (Métriques PR en attente)
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