dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

Ouverte

#3 985 ouverte le 10 juil. 2019

 (6 commentaires) (0 réaction) (2 personnes assignées)C# (1 901 forks)batch import
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

Guide contributeur