public NearestNeighborsCursor(RowCursor cursor, NearestNeighborsTransform parent, Func <int, bool> predicate, int colFeatures) { _inputCursor = cursor; _parent = parent; _trees = parent._trees; _k = parent._args.k; _getterFeatures = _inputCursor.GetGetter <VBuffer <float> >(colFeatures); _tempFeatures = new VBuffer <float>(); _distance = new VBuffer <float>(_k, new float[_k]); _idn = new VBuffer <long>(_k, new long[_k]); }
public NearestNeighborsCursor(DataViewRowCursor cursor, NearestNeighborsTransform parent, IEnumerable <DataViewSchema.Column> columnsNeeded, DataViewSchema.Column colFeatures) { _inputCursor = cursor; _parent = parent; _trees = parent._trees; _k = parent._args.k; _getterFeatures = _inputCursor.GetGetter <VBuffer <float> >(colFeatures); _tempFeatures = new VBuffer <float>(); _distance = new VBuffer <float>(_k, new float[_k]); _idn = new VBuffer <long>(_k, new long[_k]); }
public static CommonOutputs.TransformOutput NearestNeighbors(IHostEnvironment env, NearestNeighborsTransform_ArgumentsEntryPoint input) { Contracts.CheckValue(env, nameof(env)); env.CheckValue(input, nameof(input)); var h = EntryPointUtils.CheckArgsAndCreateHost(env, EP_NearestNeighbors.Name, input); var view = new _NearestNeighborsTransform(h, input, input.Data); return(new CommonOutputs.TransformOutput() { Model = new TransformModel(h, view, input.Data), OutputData = view }); }