dotnet / dotnet/machinelearning
Add FieldAwareFactorizationMachine to AutoML
Open
area-AutoML
good first issue
in-pr
Priority:2
- Dominant language
- C#
- Stars
- 9.4k
- Forks
- 2k
- Avg merge
- 2d 20h
- Merged PRs (30d)
- 11
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
Assessment
This issue has not been assessed yet.