namespace Microsoft.ML.TorchSharp
{
    public static class TorchSharpCatalog
    {
        public static SentenceSimilarityTrainer SentenceSimilarity(this RegressionTrainers catalog, string labelColumnName = "Label", string scoreColumnName = "Score", string sentence1ColumnName = "Sentence1", string sentence2ColumnName = "Sentence2", int batchSize = 32, int maxEpochs = 10, BertArchitecture architecture = BertArchitecture.Roberta, IDataView validationSet = null);
    }
}