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
});
}
