ModelOperationsCatalog.Save Метод

Определение

Пространство имен:

Microsoft.ML

Сборка:

Microsoft.ML.Data.dll

Пакет:

Microsoft.ML v2.0.0

Перегрузки

Save(ITransformer, DataViewSchema, Stream)	Сохраните модель преобразователя и схему данных, которые использовались для обучения в потоке.
Save(ITransformer, DataViewSchema, String)	Сохраните модель преобразователя и схему данных, которые использовались для его обучения в файл.
Save<TSource>(ITransformer, IDataLoader<TSource>, Stream)	Сохраните модель преобразователя и загрузчик, используемые для создания входных данных в поток.
Save<TSource>(ITransformer, IDataLoader<TSource>, String)	Сохраните модель преобразователя и загрузчик, используемый для создания входных данных в файл.

Save(ITransformer, DataViewSchema, Stream)

Сохраните модель преобразователя и схему данных, которые использовались для обучения в потоке.

public void Save (Microsoft.ML.ITransformer model, Microsoft.ML.DataViewSchema inputSchema, System.IO.Stream stream);

member this.Save : Microsoft.ML.ITransformer * Microsoft.ML.DataViewSchema * System.IO.Stream -> unit

Public Sub Save (model As ITransformer, inputSchema As DataViewSchema, stream As Stream)

Параметры

model

ITransformer

Обученная модель для сохранения. Обратите внимание, что это может быть nullсокращено для пустой цепочки преобразователей. После загрузки с Load(Stream, DataViewSchema) возвращаемым значением будет пустым TransformerChain<TLastTransformer>.

inputSchema

DataViewSchema

Схема входных данных преобразователя. Это может быть null.

stream

Stream

Доступный для записи поток для сохранения.

Save(ITransformer, DataViewSchema, String)

Сохраните модель преобразователя и схему данных, которые использовались для его обучения в файл.

public void Save (Microsoft.ML.ITransformer model, Microsoft.ML.DataViewSchema inputSchema, string filePath);

member this.Save : Microsoft.ML.ITransformer * Microsoft.ML.DataViewSchema * string -> unit

Public Sub Save (model As ITransformer, inputSchema As DataViewSchema, filePath As String)

Параметры

model

ITransformer

Обученная модель для сохранения. Обратите внимание, что это может быть nullсокращено для пустой цепочки преобразователей. После загрузки с Load(Stream, DataViewSchema) возвращаемым значением будет пустым TransformerChain<TLastTransformer>.

inputSchema

DataViewSchema

Схема входных данных преобразователя. Это может быть null.

filePath

String

Путь, по которому следует сохранить модель.

Примеры

using System;
using System.Collections.Generic;
using System.IO;
using Microsoft.ML;

namespace Samples.Dynamic.ModelOperations
{
    public class SaveLoadModel
    {
        public static void Example()
        {
            // Create a new ML context, for ML.NET operations. It can be used for
            // exception tracking and logging, as well as the source of randomness.
            var mlContext = new MLContext();

            // Generate sample data.
            var data = new List<Data>()
            {
                new Data() { Value="abc" }
            };

            // Convert data to IDataView.
            var dataView = mlContext.Data.LoadFromEnumerable(data);
            var inputColumnName = nameof(Data.Value);
            var outputColumnName = nameof(Transformation.Key);

            // Transform.
            ITransformer model = mlContext.Transforms.Conversion
                .MapValueToKey(outputColumnName, inputColumnName).Fit(dataView);

            // Save model.
            mlContext.Model.Save(model, dataView.Schema, "model.zip");

            // Load model.
            using (var file = File.OpenRead("model.zip"))
                model = mlContext.Model.Load(file, out DataViewSchema schema);

            // Create a prediction engine from the model for feeding new data.
            var engine = mlContext.Model
                .CreatePredictionEngine<Data, Transformation>(model);

            var transformation = engine.Predict(new Data() { Value = "abc" });

            // Print transformation to console.
            Console.WriteLine("Value: {0}\t Key:{1}", transformation.Value,
                transformation.Key);

            // Value: abc       Key:1

        }

        private class Data
        {
            public string Value { get; set; }
        }

        private class Transformation
        {
            public string Value { get; set; }
            public uint Key { get; set; }
        }
    }
}

Save<TSource>(ITransformer, IDataLoader<TSource>, Stream)

Сохраните модель преобразователя и загрузчик, используемые для создания входных данных в поток.

public void Save<TSource> (Microsoft.ML.ITransformer model, Microsoft.ML.IDataLoader<TSource> loader, System.IO.Stream stream);

member this.Save : Microsoft.ML.ITransformer * Microsoft.ML.IDataLoader<'Source> * System.IO.Stream -> unit

Public Sub Save(Of TSource) (model As ITransformer, loader As IDataLoader(Of TSource), stream As Stream)

Параметры типа

TSource

Параметры

model

ITransformer

Обученная модель для сохранения. Обратите внимание, что это может быть nullсокращено для пустой цепочки преобразователей. После загрузки с LoadWithDataLoader(Stream, IDataLoader<IMultiStreamSource>) возвращаемым значением будет пустым TransformerChain<TLastTransformer>.

loader

IDataLoader<TSource>

Загрузчик, который использовался для создания данных для обучения модели.

stream

Stream

Доступный для записи поток для сохранения.

Save<TSource>(ITransformer, IDataLoader<TSource>, String)

Сохраните модель преобразователя и загрузчик, используемый для создания входных данных в файл.

public void Save<TSource> (Microsoft.ML.ITransformer model, Microsoft.ML.IDataLoader<TSource> loader, string filePath);

member this.Save : Microsoft.ML.ITransformer * Microsoft.ML.IDataLoader<'Source> * string -> unit

Public Sub Save(Of TSource) (model As ITransformer, loader As IDataLoader(Of TSource), filePath As String)

Параметры типа

TSource

Параметры

model

ITransformer

Обученная модель для сохранения. Обратите внимание, что это может быть nullсокращено для пустой цепочки преобразователей. После загрузки с LoadWithDataLoader(Stream, IDataLoader<IMultiStreamSource>) возвращаемым значением будет пустым TransformerChain<TLastTransformer>.

loader

IDataLoader<TSource>

Загрузчик, который использовался для создания данных для обучения модели.

filePath

String

Путь, по которому следует сохранить модель.