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