@@ -38,6 +38,11 @@ private protected sealed class BindingsImpl : BindingsBase
38
38
public readonly int ScoreColumnIndex ;
39
39
// The type of the derived column.
40
40
public readonly DataViewType PredColType ;
41
+ /// <summary>
42
+ /// The name of the column that contains the predicted labels.
43
+ /// This field is used in the scoring process to store or reference the predicted label column.
44
+ /// </summary>
45
+ public readonly string PredictedLabelColumnName ;
41
46
// The ScoreColumnKind metadata value for all score columns.
42
47
public readonly string ScoreColumnKind ;
43
48
@@ -54,6 +59,7 @@ private BindingsImpl(DataViewSchema input, ISchemaBoundRowMapper mapper, string
54
59
ScoreColumnIndex = scoreColIndex ;
55
60
ScoreColumnKind = scoreColumnKind ;
56
61
PredColType = predColType ;
62
+ PredictedLabelColumnName = predictedLabelColumnName ;
57
63
58
64
_getScoreColumnKind = GetScoreColumnKind ;
59
65
_getScoreValueKind = GetScoreValueKind ;
@@ -113,7 +119,7 @@ public BindingsImpl ApplyToSchema(DataViewSchema input, ISchemaBindableMapper bi
113
119
bool tmp = rowMapper . OutputSchema . TryGetColumnIndex ( scoreCol , out mapperScoreColumn ) ;
114
120
env . Check ( tmp , "Mapper doesn't have expected score column" ) ;
115
121
116
- return new BindingsImpl ( input , rowMapper , Suffix , ScoreColumnKind , true , mapperScoreColumn , PredColType ) ;
122
+ return new BindingsImpl ( input , rowMapper , Suffix , ScoreColumnKind , true , mapperScoreColumn , PredColType , PredictedLabelColumnName ) ;
117
123
}
118
124
119
125
public static BindingsImpl Create ( ModelLoadContext ctx , DataViewSchema input ,
0 commit comments