@@ -128,11 +128,11 @@ private protected override void CheckScoreAndLabelTypes(RoleMappedSchema schema)
128
128
var host = Host . SchemaSensitive ( ) ;
129
129
var t = score . Type ;
130
130
if ( t != NumberType . Float )
131
- throw host . ExceptSchemaMismatch ( nameof ( schema ) , "score" , score . Name , "R4 " , t . ToString ( ) ) ;
131
+ throw host . ExceptSchemaMismatch ( nameof ( schema ) , "score" , score . Name , "float " , t . ToString ( ) ) ;
132
132
host . Check ( schema . Label . HasValue , "Could not find the label column" ) ;
133
133
t = schema . Label . Value . Type ;
134
134
if ( t != NumberType . R4 && t != NumberType . R8 && t != BoolType . Instance && t . GetKeyCount ( ) != 2 )
135
- throw host . ExceptSchemaMismatch ( nameof ( schema ) , "label" , schema . Label . Value . Name , "R4, R8, BL or a 2-value key" , t . ToString ( ) ) ;
135
+ throw host . ExceptSchemaMismatch ( nameof ( schema ) , "label" , schema . Label . Value . Name , "float, double, bool, or a 2-value key" , t . ToString ( ) ) ;
136
136
}
137
137
138
138
private protected override void CheckCustomColumnTypesCore ( RoleMappedSchema schema )
@@ -144,7 +144,7 @@ private protected override void CheckCustomColumnTypesCore(RoleMappedSchema sche
144
144
host . CheckParam ( prob . Count == 1 , nameof ( schema ) , "Cannot have multiple probability columns" ) ;
145
145
var probType = prob [ 0 ] . Type ;
146
146
if ( probType != NumberType . Float )
147
- throw host . ExceptSchemaMismatch ( nameof ( schema ) , "probability" , prob [ 0 ] . Name , "R4 " , probType . ToString ( ) ) ;
147
+ throw host . ExceptSchemaMismatch ( nameof ( schema ) , "probability" , prob [ 0 ] . Name , "float " , probType . ToString ( ) ) ;
148
148
}
149
149
else if ( ! _useRaw )
150
150
{
0 commit comments