Skip to content

Commit e95f31d

Browse files
MarcinJuraszekIvanidzo4ka
authored andcommitted
Pass weighting value from ProduceWordBags to WordBagEstimator constructor (#3135)
1 parent 617f0f6 commit e95f31d

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
@@ -297,7 +297,7 @@ public static WordBagEstimator ProduceWordBags(this TransformsCatalog.TextTransf
297297
int maximumNgramsCount = NgramExtractingEstimator.Defaults.MaximumNgramsCount,
298298
NgramExtractingEstimator.WeightingCriteria weighting = NgramExtractingEstimator.WeightingCriteria.Tf)
299299
=> new WordBagEstimator(Contracts.CheckRef(catalog, nameof(catalog)).GetEnvironment(),
300-
outputColumnName, inputColumnName, ngramLength, skipLength, useAllLengths, maximumNgramsCount);
300+
outputColumnName, inputColumnName, ngramLength, skipLength, useAllLengths, maximumNgramsCount, weighting);
301301

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

0 commit comments

Comments
 (0)