Method Transform

catboost/dotnet/CatBoostNet/CatBoostModel.cs:55–202

(IDataView input)

        /// <inheritdoc />
        public IDataView Transform(IDataView input)
        {
            HashSet<string> numTypes = new HashSet<string>
            {
                "Boolean", "Single", "Int32"
            };

            int nFloat = 0, nCat = 0;
            Collection<int> catFeatureIxs = new Collection<int>();

            using (var cursor = input.GetRowCursor(input.Schema))
            {
                cursor.MoveNext();
                int ix = -1;
                int ptr = 0;
                foreach (var col in cursor.Schema)
                {
                    ix++;
                    if (col.Name == TargetFeature)
                    {
                        continue;
                    }

                    string typeId = col.Type.ToString();
                    if (numTypes.Contains(typeId))
                    {
                        nFloat++;
                    }
                    else if (typeId == "String")
                    {
                        catFeatureIxs.Add(ptr);
                        nCat++;
                    }
                    else
                    {
                        throw new NotSupportedException($"Data type {typeId} is not supported.");
                    }
                    ptr++;
                }
            }

            List<float[]> floatFeatures = new List<float[]>();
            List<string[]> catFeatures = new List<string[]>();

            using (var cursor = input.GetRowCursor(input.Schema))
            {
                int rowIx = 0;
                while (cursor.MoveNext())
                {
                    int ix = -1;
                    int ptr = 0;
                    int realPtr = 0;
                    int catPtr = 0;

                    float[] floats = new float[nFloat];
                    string[] cats = new string[nCat];

                    foreach (var col in cursor.Schema)