@@ -26,9 +26,8 @@ public static CommonOutputs.TransformOutput SelectColumns(IHostEnvironment env,
26
26
Contracts . CheckValue ( env , nameof ( env ) ) ;
27
27
env . CheckValue ( input , nameof ( input ) ) ;
28
28
EntryPointUtils . CheckInputArgs ( env , input ) ;
29
- int colMax ;
30
29
var view = input . Data ;
31
- var maxScoreId = view . Schema . GetMaxMetadataKind ( out colMax , MetadataUtils . Kinds . ScoreColumnSetId ) ;
30
+ var maxScoreId = view . Schema . GetMaxMetadataKind ( out int colMax , MetadataUtils . Kinds . ScoreColumnSetId ) ;
32
31
List < int > indices = new List < int > ( ) ;
33
32
for ( int i = 0 ; i < view . Schema . ColumnCount ; i ++ )
34
33
{
@@ -82,7 +81,7 @@ public static CommonOutputs.TransformOutput RenameBinaryPredictionScoreColumns(I
82
81
// Rename all the score columns.
83
82
int colMax ;
84
83
var maxScoreId = input . Data . Schema . GetMaxMetadataKind ( out colMax , MetadataUtils . Kinds . ScoreColumnSetId ) ;
85
- var copyCols = new List < CopyColumnsTransform . Column > ( ) ;
84
+ var copyCols = new List < ( string Source , string Name ) > ( ) ;
86
85
for ( int i = 0 ; i < input . Data . Schema . ColumnCount ; i ++ )
87
86
{
88
87
if ( input . Data . Schema . IsHidden ( i ) )
@@ -99,10 +98,10 @@ public static CommonOutputs.TransformOutput RenameBinaryPredictionScoreColumns(I
99
98
}
100
99
var source = input . Data . Schema . GetColumnName ( i ) ;
101
100
var name = source + "." + positiveClass ;
102
- copyCols . Add ( new CopyColumnsTransform . Column ( ) { Name = name , Source = source } ) ;
101
+ copyCols . Add ( ( source , name ) ) ;
103
102
}
104
103
105
- var copyColumn = new CopyColumnsTransform ( env , new CopyColumnsTransform . Arguments ( ) { Column = copyCols . ToArray ( ) } , input . Data ) ;
104
+ var copyColumn = new CopyColumnsTransform ( env , copyCols . ToArray ( ) ) . Transform ( input . Data ) ;
106
105
var dropColumn = new DropColumnsTransform ( env , new DropColumnsTransform . Arguments ( ) { Column = copyCols . Select ( c => c . Source ) . ToArray ( ) } , copyColumn ) ;
107
106
return new CommonOutputs . TransformOutput { Model = new TransformModel ( env , dropColumn , input . Data ) , OutputData = dropColumn } ;
108
107
}
0 commit comments