dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

オープン

#3,985 opened on 2019/07/10

 (6 件のコメント) (0 件のリアクション) (2 人の担当者)C# (1,901 件のフォーク)batch import
AutoML.NETP2command-linegood first issuein-pr

Repository metrics

Stars
 (9,125 個のスター)
PR merge metrics
 (PR metrics pending)

説明

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

コントリビューターガイド