dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

Open

Aperta il 10 lug 2019

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

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

Tech stack
csharp
Dominio
machine learning
Tipo issue
feature
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 hours
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
blocked
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
clear
Prerequisiti
C#ML.NET AutoML
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
35
Direzione di ricerca
Refer to the linked file src/Microsoft.ML.AutoML/TrainerExtensions/BinaryTrainerExtensions.cs for the pattern used by SDCA. Add a new entry for FieldAwareFactorizationMachine similar to SDCA. The sweep range and CLI CodeGen changes should follow the same pattern. Since there are assignees, discuss progress with them.