public void XGBoostTreeConverter_ParseNodeIndex()
{
var actual = XGBoostTreeConverter.ParseNodeIndex("1:[f2<2.695] yes=1,no=2,missing=1,gain=343.922,cover=214");
var expected = 1;
Assert.AreEqual(expected, actual);
}
public void XGBoostTreeConverter_ParseSplitValue()
{
var actual = XGBoostTreeConverter.ParseSplitValue("0:[f2<2.695] yes=1,no=2,missing=1,gain=343.922,cover=214");
var expected = 2.695;
Assert.AreEqual(expected, actual, m_delta);
}
public void XGBoostTreeConverter_ParseLeafValue()
{
var actual = XGBoostTreeConverter.ParseLeafValue("7:leaf=0.404167,cover=35");
var expected = 0.404167;
Assert.AreEqual(expected, actual, m_delta);
}
public void XGBoostTreeConverter_ConvertXGBoostTextTreeToGBMTree_Tree_2()
{
var actualNodes = XGBoostTreeConverter.ConvertXGBoostTextTreeToGBMTree(m_tree2).Nodes;
var expectedNodes = m_tree2Nodes;
Assert.AreEqual(expectedNodes.Count, actualNodes.Count);
for (int i = 0; i < expectedNodes.Count; i++)
{
var expected = expectedNodes[i];
var actual = actualNodes[i];
AssertGBMNode(expected, actual);
}
}
public void XGBoostTreeConverter_IsLeaf()
{
Assert.IsTrue(XGBoostTreeConverter.IsLeaf("7:leaf=0.404167,cover=35"));
Assert.IsFalse(XGBoostTreeConverter.IsLeaf("0:[f2<2.695] yes=1,no=2,missing=1,gain=343.922,cover=214"));
}
