dotnet/machinelearning

Add FieldAwareFactorizationMachine to AutoML

Open

#3,985 opened on Jul 10, 2019

View on GitHub
 (6 comments) (0 reactions) (2 assignees)C# (9,125 stars) (1,901 forks)batch import
AutoML.NETP2command-linegood first issue

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

Contributor guide

Tech stack
csharp
Domain
machine learning
Issue type
feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
2
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-3 hours
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
blocked
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
clear
Prerequisites
C#ML.NET AutoML
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
35
Research direction
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.