dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

Offen

#3.985 geöffnet am 10.07.2019

 (6 Kommentare) (0 Reaktionen) (2 zugewiesene Personen)C# (1.901 Forks)batch import
AutoML.NETP2command-linegood first issuein-pr

Repository-Metriken

Stars
 (9.125 Sterne)
PR-Merge-Metriken
 (PR-Metriken ausstehend)

Beschreibung

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

Contributor Guide