dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

Open

#3,985 建立於 2019年7月10日

在 GitHub 查看
 (6 留言) (0 反應) (2 負責人)C# (1,901 fork)batch import
AutoML.NETP2command-linegood first issue

倉庫指標

Star
 (9,125 star)
PR 合併指標
 (平均合併 8天 1小時) (30 天內合併 3 個 PR)

描述

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

貢獻者指南