dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

Open

#3.985 aberto em 10 de jul. de 2019

Ver no GitHub
 (6 comments) (0 reactions) (2 assignees)C# (1.901 forks)batch import
AutoML.NETP2command-linegood first issue

Métricas do repositório

Stars
 (9.125 stars)
Métricas de merge de PR
 (Mesclagem média 6d 19h) (5 fundiu PRs em 30d)

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

Guia do colaborador