Skip to content

Commit 349b2c0

Browse files
MarcinJuraszekShauheen
authored andcommitted
Pass weighting value from ProduceWordBags to WordBagEstimator constructor (dotnet#3135)
1 parent 43f2eb2 commit 349b2c0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/Microsoft.ML.Transforms/Text/TextCatalog.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ public static WordBagEstimator ProduceWordBags(this TransformsCatalog.TextTransf
283283
int maximumNgramsCount = NgramExtractingEstimator.Defaults.MaximumNgramsCount,
284284
NgramExtractingEstimator.WeightingCriteria weighting = NgramExtractingEstimator.WeightingCriteria.Tf)
285285
=> new WordBagEstimator(Contracts.CheckRef(catalog, nameof(catalog)).GetEnvironment(),
286-
outputColumnName, inputColumnName, ngramLength, skipLength, useAllLengths, maximumNgramsCount);
286+
outputColumnName, inputColumnName, ngramLength, skipLength, useAllLengths, maximumNgramsCount, weighting);
287287

288288
/// <summary>
289289
/// Produces a bag of counts of ngrams (sequences of consecutive words) in <paramref name="inputColumnNames"/>

0 commit comments

Comments
 (0)