dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

Open

#3985 aperta il 10 lug 2019

Vedi su GitHub
 (6 commenti) (0 reazioni) (2 assegnatari)C# (1901 fork)batch import
AutoML.NETP2command-linegood first issuein-pr

Metriche repository

Star
 (9125 star)
Metriche merge PR
 (Merge medio 6g 19h) (5 PR mergiate in 30 g)

Descrizione

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

Guida contributor