@@ -20,14 +20,16 @@ internal class EmptyWriter : TextWriter
20
20
public override Encoding Encoding => null ;
21
21
}
22
22
23
- public class BigramAndTrigramBenchmark
23
+ public class MultiClassClassification
24
24
{
25
25
private string _dataPath_Wiki ;
26
26
private string _modelPath_Wiki ;
27
27
28
28
[ GlobalSetup ( Targets = new string [ ] {
29
29
nameof ( CV_Multiclass_WikiDetox_BigramsAndTrichar_OVAAveragedPerceptron ) ,
30
- nameof ( CV_Multiclass_WikiDetox_BigramsAndTrichar_LightGBMMulticlass ) } ) ]
30
+ nameof ( CV_Multiclass_WikiDetox_BigramsAndTrichar_LightGBMMulticlass ) ,
31
+ nameof ( CV_Multiclass_WikiDetox_WordEmbeddings_OVAAveragedPerceptron ) ,
32
+ nameof ( CV_Multiclass_WikiDetox_WordEmbeddings_SDCAMC ) } ) ]
31
33
public void SetupTrainingSpeedTests ( )
32
34
{
33
35
_dataPath_Wiki = Path . GetFullPath ( TestDatasets . WikiDetox . trainFilename ) ;
@@ -81,5 +83,25 @@ public void Test_Multiclass_WikiDetox_BigramsAndTrichar_OVAAveragedPerceptron()
81
83
Maml . MainCore ( tlc , cmd , alwaysPrintStacktrace : false ) ;
82
84
}
83
85
}
86
+
87
+ [ Benchmark ]
88
+ public void CV_Multiclass_WikiDetox_WordEmbeddings_OVAAveragedPerceptron ( )
89
+ {
90
+ string cmd = @"CV tr=OVA{p=AveragedPerceptron{iter=10}} k=5 loader=TextLoader{quote=- sparse=- col=Label:R4:0 col=rev_id:TX:1 col=comment:TX:2 col=logged_in:BL:4 col=ns:TX:5 col=sample:TX:6 col=split:TX:7 col=year:R4:3 header=+} data=" + _dataPath_Wiki + " xf=Convert{col=logged_in type=R4} xf=CategoricalTransform{col=ns} xf=TextTransform{col=FeaturesText:comment tokens=+ wordExtractor=NGramExtractorTransform{ngram=2}} xf=WordEmbeddingsTransform{col=FeaturesWordEmbedding:FeaturesText_TransformedText model=FastTextWikipedia300D} xf=Concat{col=Features:FeaturesText,FeaturesWordEmbedding,logged_in,ns}" ;
91
+ using ( var tlc = new TlcEnvironment ( verbose : false , sensitivity : MessageSensitivity . None , outWriter : EmptyWriter . Instance ) )
92
+ {
93
+ Maml . MainCore ( tlc , cmd , alwaysPrintStacktrace : false ) ;
94
+ }
95
+ }
96
+
97
+ [ Benchmark ]
98
+ public void CV_Multiclass_WikiDetox_WordEmbeddings_SDCAMC ( )
99
+ {
100
+ string cmd = @"CV tr=SDCAMC k=5 loader=TextLoader{quote=- sparse=- col=Label:R4:0 col=rev_id:TX:1 col=comment:TX:2 col=logged_in:BL:4 col=ns:TX:5 col=sample:TX:6 col=split:TX:7 col=year:R4:3 header=+} data=" + _dataPath_Wiki + " xf=Convert{col=logged_in type=R4} xf=CategoricalTransform{col=ns} xf=TextTransform{col=FeaturesText:comment tokens=+ wordExtractor={} charExtractor={}} xf=WordEmbeddingsTransform{col=FeaturesWordEmbedding:FeaturesText_TransformedText model=FastTextWikipedia300D} xf=Concat{col=Features:FeaturesWordEmbedding,logged_in,ns}" ;
101
+ using ( var tlc = new TlcEnvironment ( verbose : false , sensitivity : MessageSensitivity . None , outWriter : EmptyWriter . Instance ) )
102
+ {
103
+ Maml . MainCore ( tlc , cmd , alwaysPrintStacktrace : false ) ;
104
+ }
105
+ }
84
106
}
85
107
}
0 commit comments