public void CanHandleUtf8()
{
using var fastText = new FastTextWrapper(loggerFactory: _loggerFactory);
string outPath = Path.Combine(_tempDir, "rus");
fastText.Supervised("data.rus.txt", outPath, new SupervisedArgs());
var labels = fastText.GetLabels();
labels.Length.Should().Be(2);
labels.Should().Contain(new[] { "__label__оператор", "__label__выход" });
var pred = fastText.PredictSingle("Позови оператора");
pred.Probability.Should().BeGreaterThan(0);
pred.Label.Should().Be("__label__оператор");
var sourceWords = File.ReadAllText("data.rus.txt")
.Split(new[] { " ", "\r\n", "\n" }, StringSplitOptions.RemoveEmptyEntries)
.Where(x => !x.StartsWith("__label__"))
.Distinct().ToArray();
var nn = fastText.GetNearestNeighbours("оператор", 2);
nn.Length.Should().Be(2);
sourceWords.Should().Contain(nn.Select(x => x.Label));
foreach (var prediction in nn)
{
prediction.Probability.Should().BeGreaterThan(0);
}
}
public void CanTrainSupervised()
{
using var fastText = new FastTextWrapper(loggerFactory: _loggerFactory);
string outPath = Path.Combine(_tempDir, "cooking");
var args = new SupervisedArgs();
var tuneArgs = new AutotuneArgs();
fastText.Supervised("cooking.train.txt", outPath, args, tuneArgs, true);
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(100);
fastText.ModelPath.Should().Be(outPath + ".bin");
AssertLabels(fastText.GetLabels());
File.Exists(outPath + ".bin").Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
var debugArgs = DebugArgs.Load("_train.txt");
AssertSupervisedArgs(args, debugArgs.ExternalArgs);
AssertSupervisedArgs(args, debugArgs.ConvertedArgs);
AssertAutotuneArgs(tuneArgs, debugArgs.ExternalTune);
AssertAutotuneArgs(tuneArgs, debugArgs.ConvertedTune);
debugArgs.ExternalInput.Should().Be("cooking.train.txt");
debugArgs.ConvertedInput.Should().Be("cooking.train.txt");
debugArgs.ExternalOutput.Should().Be(outPath);
debugArgs.ConvertedOutput.Should().Be(outPath);
}
public void CanTrainSupervisedWithProgressCallback()
{
using var fastText = new FastTextWrapper();
string outPath = Path.Combine(_tempDir, "cooking");
int callNum = 0;
var args = new SupervisedArgs
{
TrainProgressCallback = (progress, loss, wst, lr, eta) =>
{
callNum++;
}
};
fastText.Supervised("cooking.train.txt", outPath, args);
callNum.Should().BeGreaterThan(0);
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(100);
fastText.ModelPath.Should().Be(outPath + ".bin");
AssertLabels(fastText.GetLabels());
File.Exists(outPath + ".bin").Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
}
static void Main(string[] args)
{
Log.Logger = new LoggerConfiguration()
.MinimumLevel.Debug()
.WriteTo.Console(theme: ConsoleTheme.None)
.CreateLogger();
var log = Log.ForContext <Program>();
var tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N"));
Directory.CreateDirectory(tempDir);
log.Information($"Temp dir: {tempDir}");
string outPath = Path.Combine(tempDir, "cooking.bin");
var fastText = new FastTextWrapper(loggerFactory: new LoggerFactory(new[] { new SerilogLoggerProvider() }));
var ftArgs = FastTextArgs.SupervisedDefaults();
ftArgs.epoch = 15;
ftArgs.lr = 1;
ftArgs.dim = 300;
ftArgs.wordNgrams = 2;
ftArgs.minn = 3;
ftArgs.maxn = 6;
fastText.Supervised("cooking.train.txt", outPath, ftArgs);
}
public void CanAutotuneSupervisedModel()
{
using var fastText = new FastTextWrapper(loggerFactory: _loggerFactory);
string outPath = Path.Combine(_tempDir, "cooking");
var args = new SupervisedArgs
{
bucket = 2100000,
dim = 250,
epoch = 10,
loss = LossName.HierarchicalSoftmax,
lr = 0.5,
maxn = 5,
minn = 2,
neg = 6,
seed = 42,
t = 0.0002,
thread = 10,
verbose = 1,
ws = 6,
minCount = 2,
saveOutput = true,
wordNgrams = 2,
lrUpdateRate = 110,
minCountLabel = 1
};
var autotuneArgs = new AutotuneArgs
{
Duration = 30,
Metric = "precisionAtRecall:30",
Predictions = 2,
ValidationFile = "cooking.valid.txt"
};
fastText.Supervised("cooking.train.txt", outPath, args, autotuneArgs, true);
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(250);
fastText.ModelPath.Should().Be(outPath + ".bin");
File.Exists(outPath + ".bin").Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
var debugArgs = DebugArgs.Load("_train.txt");
AssertSupervisedArgs(args, debugArgs.ExternalArgs);
AssertSupervisedArgs(args, debugArgs.ConvertedArgs);
AssertAutotuneArgs(autotuneArgs, debugArgs.ExternalTune);
AssertAutotuneArgs(autotuneArgs, debugArgs.ConvertedTune);
debugArgs.ExternalInput.Should().Be("cooking.train.txt");
debugArgs.ConvertedInput.Should().Be("cooking.train.txt");
debugArgs.ExternalOutput.Should().Be(outPath);
debugArgs.ConvertedOutput.Should().Be(outPath);
}
static void Main(string[] args)
{
Log.Logger = new LoggerConfiguration()
.MinimumLevel.Debug()
.WriteTo.Console(theme: ConsoleTheme.None)
.CreateLogger();
var log = Log.ForContext <Program>();
var tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N"));
Directory.CreateDirectory(tempDir);
log.Information($"Temp dir: {tempDir}");
string outPath = Path.Combine(tempDir, "cooking.bin");
var fastText = new FastTextWrapper(loggerFactory: new LoggerFactory(new[] { new SerilogLoggerProvider() }));
var ftArgs = FastTextArgs.SupervisedDefaults();
ftArgs.epoch = 15;
ftArgs.lr = 1;
ftArgs.dim = 300;
ftArgs.wordNgrams = 2;
ftArgs.minn = 3;
ftArgs.maxn = 6;
fastText.Supervised("cooking.train.txt", outPath, ftArgs);
try
{
File.Delete("_debug.txt");
}
catch
{
}
var result = fastText.TestInternal("cooking.valid.txt", 1, 0.0f, true);
log.Information($"Results:\n\tPrecision: {result.GlobalMetrics.GetPrecision()}" +
$"\n\tRecall: {result.GlobalMetrics.GetRecall()}" +
$"\n\tF1: {result.GlobalMetrics.GetF1()}");
var curve = result.GetPrecisionRecallCurve();
var(_, debugCurve) = TestResult.LoadDebugResult("_debug.txt", fastText.GetLabels());
string plotPath = PlotCurves(tempDir, new [] { curve, debugCurve });
log.Information($"Precision-Recall plot: {plotPath}");
Console.WriteLine("\nPress any key to exit.");
Console.ReadKey();
Directory.Delete(tempDir, true);
}
public SupervisedModelFixture()
{
_tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N"));
Directory.CreateDirectory(_tempDir);
FastText = new FastTextWrapper();
string outPath = Path.Combine(_tempDir, "cooking");
FastText.Supervised("cooking.train.txt", outPath, new SupervisedArgs());
FastText.IsModelReady().Should().BeTrue();
File.Exists(FastText.ModelPath).Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
}
public void CanTrainSupervised()
{
var fastText = new FastTextWrapper(loggerFactory: _loggerFactory);
string outPath = Path.Combine(_tempDir, "cooking");
fastText.Supervised("cooking.train.txt", outPath, FastTextArgs.SupervisedDefaults());
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(100);
CheckLabels(fastText.GetLabels());
File.Exists(outPath + ".bin").Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
}
public void CanTrainSupervisedWithNoLoggingAndNoArgs()
{
using var fastText = new FastTextWrapper();
string outPath = Path.Combine(_tempDir, "cooking");
fastText.Supervised("cooking.train.txt", outPath);
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(100);
fastText.ModelPath.Should().Be(outPath + ".bin");
AssertLabels(fastText.GetLabels());
File.Exists(outPath + ".bin").Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
}
public void CanUsePretrainedVectorsForSupervisedModel()
{
var fastText = new FastTextWrapper(loggerFactory: _loggerFactory);
string outPath = Path.Combine(_tempDir, "cooking");
var args = FastTextArgs.SupervisedDefaults();
args.PretrainedVectors = "cooking.unsup.300.vec";
args.dim = 300;
fastText.Supervised("cooking.train.txt", outPath, args);
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(300);
CheckLabels(fastText.GetLabels());
File.Exists(outPath + ".bin").Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
}
public void CanTrainSupervisedWithPretrainedVectors()
{
using var fastText = new FastTextWrapper(loggerFactory: _loggerFactory);
string outPath = Path.Combine(_tempDir, "cooking");
var args = new SupervisedArgs();
args.PretrainedVectors = "cooking.unsup.300.vec";
args.dim = 300;
fastText.Supervised("cooking.train.txt", outPath, args, new AutotuneArgs(), true);
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(300);
fastText.ModelPath.Should().Be(outPath + ".bin");
AssertLabels(fastText.GetLabels());
File.Exists(outPath + ".bin").Should().BeTrue();
File.Exists(outPath + ".vec").Should().BeTrue();
}
public void CanTrainSupervisedWithRelativeOutput()
{
using var fastText = new FastTextWrapper(loggerFactory: _loggerFactory);
var args = new SupervisedArgs();
var tuneArgs = new AutotuneArgs();
fastText.Supervised("cooking.train.txt", "cooking", args, tuneArgs, true);
fastText.IsModelReady().Should().BeTrue();
fastText.GetModelDimension().Should().Be(100);
fastText.ModelPath.Should().Be("cooking.bin");
AssertLabels(fastText.GetLabels());
File.Exists("cooking.bin").Should().BeTrue();
File.Exists("cooking.vec").Should().BeTrue();
File.Delete("cooking.bin");
File.Delete("cooking.vec");
}
static void Main(string[] args)
{
Log.Logger = new LoggerConfiguration()
.MinimumLevel.Debug()
.WriteTo.Console(theme: ConsoleTheme.None)
.CreateLogger();
var log = Log.ForContext <Program>();
var tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N"));
Directory.CreateDirectory(tempDir);
log.Information($"Temp dir: {tempDir}");
string outPath = Path.Combine(tempDir, "cooking.bin");
var fastText = new FastTextWrapper(loggerFactory: new LoggerFactory(new[] { new SerilogLoggerProvider() }));
AnsiConsole.Progress()
.Start(ctx =>
{
var task = ctx.AddTask("Training");
var ftArgs = new SupervisedArgs
{
epoch = 15,
lr = 1,
dim = 300,
wordNgrams = 2,
minn = 3,
maxn = 6,
verbose = 0,
TrainProgressCallback = (progress, loss, wst, lr, eta) =>
{
task.Value = Math.Ceiling(progress * 100);
task.Description = $"Loss: {loss:N3}, words/thread/sec: {wst}, LR: {lr:N5}, ETA: {eta}";
}
};
fastText.Supervised("cooking.train.txt", outPath, ftArgs);
});
try
{
File.Delete("_debug.txt");
}
catch
{
}
log.Information("Validating model on the test set");
var result = fastText.TestInternal("cooking.valid.txt", 1, 0.0f, true);
log.Information($"Results:\n\tPrecision: {result.GlobalMetrics.GetPrecision()}" +
$"\n\tRecall: {result.GlobalMetrics.GetRecall()}" +
$"\n\tF1: {result.GlobalMetrics.GetF1()}");
var curve = result.GetPrecisionRecallCurve();
var(_, debugCurve) = TestResult.LoadDebugResult("_debug.txt", fastText.GetLabels());
string plotPath = PlotCurves(tempDir, new [] { curve, debugCurve });
log.Information($"Precision-Recall plot: {plotPath}");
Console.WriteLine("\nPress any key to exit.");
Console.ReadKey();
Directory.Delete(tempDir, true);
}
