private SortInDataFrameTransform(IHost host, ModelLoadContext ctx, IDataView input) : base(host, input) { Host.CheckValue(input, "input"); Host.CheckValue(ctx, "ctx"); _sortColumn = ctx.Reader.ReadString(); Host.AssertValue(_sortColumn); int index; var schema = input.Schema; if (!schema.TryGetColumnIndex(_sortColumn, out index)) { Contracts.Check(false, "sortColumn not found in input schema."); } var type = schema.GetColumnType(index); Host.Check(!type.IsVector(), "sortColumn cannot be a vector."); _reverse = ctx.Reader.ReadBoolean(); _numThreads = ctx.Reader.ReadInt32(); if (_numThreads < 0) { _numThreads = null; } _transform = CreateTemplatedTransform(); }
public static SortInDataFrameTransform Create(IHostEnvironment env, ModelLoadContext ctx, IDataView input) { Contracts.CheckValue(env, "env"); var h = env.Register(RegistrationName); h.CheckValue(ctx, "ctx"); h.CheckValue(input, "input"); ctx.CheckAtModel(GetVersionInfo()); return(h.Apply("Loading Model", ch => new SortInDataFrameTransform(h, ctx, input))); }
private ExtendedCacheTransform(IHost host, ModelLoadContext ctx, IDataView input) : base(host, input) { Host.CheckValue(input, "input"); Host.CheckValue(ctx, "ctx"); _inDataFrame = ctx.Reader.ReadBoolean(); _async = ctx.Reader.ReadBoolean(); _numThreads = ctx.Reader.ReadInt32(); host.Check(_numThreads > -2, "_numThreads"); if (_numThreads < 0) { _numThreads = null; } _saverSettings = ctx.Reader.ReadString(); if (_inDataFrame) { _cacheFile = null; _reuse = false; } else { _cacheFile = ctx.Reader.ReadString(); _reuse = ctx.Reader.ReadBoolean(); host.CheckValue(_cacheFile, "_cacheFile"); } var saver = ComponentCreation.CreateSaver(Host, _saverSettings); if (saver == null) { throw Host.Except("Cannot parse '{0}'", _saverSettings); } _pipedTransform = CreatePipeline(host, input); }