-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Modify API for advanced settings (FieldAwareFactorizationMachineTrainer) #2219
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Modify API for advanced settings (FieldAwareFactorizationMachineTrainer) #2219
Conversation
s.LatentDim = 7; | ||
}); | ||
var ffmArgs = new FieldAwareFactorizationMachineTrainer.Options { | ||
FeatureColumn = "Feature1", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
FeatureColumn = "Feature1", | |
FeatureColumn = "Feature1", // Features from the 1st field. | |
``` #Resolved |
}); | ||
var ffmArgs = new FieldAwareFactorizationMachineTrainer.Options { | ||
FeatureColumn = "Feature1", | ||
ExtraFeatureColumns = new[] { "Feature2", "Feature3", "Feature4" }, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ExtraFeatureColumns = new[] { "Feature2", "Feature3", "Feature4" }, | |
ExtraFeatureColumns = new[] { "Feature2", "Feature3", "Feature4" }, // 2nd field's feature column, 3rd field's feature column, 4th field's feature column.``` #Resolved |
/// <param name="onFit">A delegate that is called every time the | ||
/// <see cref="Estimator{TInShape, TOutShape, TTransformer}.Fit(DataView{TInShape})"/> method is called on the | ||
/// <see cref="Estimator{TInShape, TOutShape, TTransformer}"/> instance created out of this. This delegate will receive | ||
/// the model that was trained. Note that this action cannot change the result in any way; it is only a way for the caller to |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What's the type of the model that was trained
? Is it possible to make a reference here? #Resolved
/// The settings here will override the ones provided in the direct method signature, | ||
/// if both are present and have different values. | ||
/// The columns names, however need to be provided directly, not through the <paramref name="advancedSettings"/>.</param>/// <param name="onFit">A delegate that is called every time the | ||
/// <param name="onFit">A delegate that is called every time the |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could we reference the type of the trained model? It's impossible for user to write onFit without knowing its input type. Many thanks. #Resolved
Codecov Report
@@ Coverage Diff @@
## master #2219 +/- ##
==========================================
+ Coverage 69.78% 80.8% +11.01%
==========================================
Files 786 159 -627
Lines 143945 28504 -115441
Branches 16648 1909 -14739
==========================================
- Hits 100451 23032 -77419
+ Misses 38951 5175 -33776
+ Partials 4543 297 -4246
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🚀
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🎩
Towards #1798 .
This PR addresses the following algos
The following changes have been made:
public
extension methods, one for simple arguments and the other for advanced optionsinternal
. Also a few other fields have been madeinternal
Options
for API consistency.