Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

First example of c# wrapper done (marshalling prediction to submission file)

Browse Source
This commit is contained in:
giuliohome 2014-08-30 03:05:40 +02:00
parent 8b26cba148
commit 2587da5fea
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
windows/xgboost_sharp_wrapper/xgboost.cs
View File

@ -50,7 +50,18 @@ public IntPtr SharpXGDMatrixCreateFromFile(string fname, int silent)
* \param len used to store length of returning result * \param len used to store length of returning result
*/ */
[DllImport(dll_path + "xgboost_wrapper.dll")] [DllImport(dll_path + "xgboost_wrapper.dll")]
public static extern Double[] XGBoosterPredict(IntPtr handle, IntPtr dmat, int output_margin, System.UInt32 len); public static extern IntPtr XGBoosterPredict(IntPtr handle, IntPtr dmat, int output_margin, ref System.UInt32 len);
public float[] SharpXGBoosterPredict(IntPtr handle, IntPtr dmat, int output_margin, System.UInt32 len)
{
IntPtr buf = XGBoosterPredict(handle, dmat, output_margin, ref len);
float[] buffer = new float[len];
Marshal.Copy(buf, buffer, 0, buffer.Length);
return buffer;
}
/*! /*!
* \brief load model from existing file * \brief load model from existing file
* \param handle handle * \param handle handle