dotnet/machinelearning
Add FieldAwareFactorizationMachine to AutoML
Aberta
#3.985 aberto em 10 de jul. de 2019
AutoML.NETP2command-linegood first issuein-pr
Métricas do repositório
- Stars
- (9.125 estrelas)
- Métricas de merge de PR
- (Métricas PR pendentes)
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