public void Test_Ranking_MSLRWeb10K_RawNumericFeatures_FastTreeRanking()
{
// This benchmark is profiling bulk scoring speed and not training speed.
string cmd = @"Test data=" + _mslrWeb10kTest + " in=" + _modelPathMslr;
var environment = EnvironmentFactory.CreateRankingEnvironment <RankingEvaluator, TextLoader, HashingTransformer, FastTreeRankingTrainer, FastTreeRankingModelParameters>();
cmd.ExecuteMamlCommand(environment);
}
public void TrainTest_Ranking_MSLRWeb10K_RawNumericFeatures_FastTreeRanking()
{
string cmd = @"TrainTest test=" + _mslrWeb10kValidate +
" eval=RankingEvaluator{t=10}" +
" data=" + _mslrWeb10kTrain +
" loader=TextLoader{col=Label:R4:0 col=GroupId:TX:1 col=Features:R4:2-138}" +
" xf=HashTransform{col=GroupId} xf=NAHandleTransform{col=Features}" +
" tr=FastTreeRanking{}";
var environment = EnvironmentFactory.CreateRankingEnvironment <RankingEvaluator, TextLoader, HashingTransformer, FastTreeRankingTrainer, FastTreeRankingModelParameters>();
cmd.ExecuteMamlCommand(environment);
}
public void SetupScoringSpeedTests()
{
_mslrWeb10kTest = GetBenchmarkDataPathAndEnsureData(TestDatasets.MSLRWeb.testFilename, TestDatasets.MSLRWeb.path);
_mslrWeb10kValidate = GetBenchmarkDataPathAndEnsureData(TestDatasets.MSLRWeb.validFilename, TestDatasets.MSLRWeb.path);
_mslrWeb10kTrain = GetBenchmarkDataPathAndEnsureData(TestDatasets.MSLRWeb.trainFilename, TestDatasets.MSLRWeb.path);
if (!File.Exists(_mslrWeb10kTest))
{
throw new FileNotFoundException(string.Format(Errors.DatasetNotFound, _mslrWeb10kTest));
}
if (!File.Exists(_mslrWeb10kValidate))
{
throw new FileNotFoundException(string.Format(Errors.DatasetNotFound, _mslrWeb10kValidate));
}
if (!File.Exists(_mslrWeb10kTrain))
{
throw new FileNotFoundException(string.Format(Errors.DatasetNotFound, _mslrWeb10kTrain));
}
_modelPathMslr = Path.Combine(Path.GetDirectoryName(typeof(RankingTest).Assembly.Location), "FastTreeRankingModel.zip");
string cmd = @"TrainTest test=" + _mslrWeb10kValidate +
" eval=RankingEvaluator{t=10}" +
" data=" + _mslrWeb10kTrain +
" loader=TextLoader{col=Label:R4:0 col=GroupId:TX:1 col=Features:R4:2-138}" +
" xf=HashTransform{col=GroupId}" +
" xf=NAHandleTransform{col=Features}" +
" tr=FastTreeRanking{}" +
" out={" + _modelPathMslr + "}";
var environment = EnvironmentFactory.CreateRankingEnvironment <RankingEvaluator, TextLoader, HashingTransformer, FastTreeRankingTrainer, FastTreeRankingModelParameters>();
cmd.ExecuteMamlCommand(environment);
}
