package MachineLearning; import java.util.*; /** * Klasa reprezentująca liść drzewa decyzyjnego * @author Angello * */ public class TermostatLeaf { protected Integer m_category; protected ArrayList m_P; /** * Konstruktor * @param category * @param P */ public TermostatLeaf(Integer category, ArrayList P) { m_category = category; this.p( P); } public TermostatLeaf() { } /** * Zwraca kategorię dla danego liścia * @return Indeks kategorii */ public Integer category() { return m_category; } /** * Ustawia kategorię danego węzła * @param category */ public void category(Integer category) { m_category = category; } public ArrayList p() { return m_P; } public void p(ArrayList p) { m_P = new ArrayList( p); } /** * Zwraca kategorię * @param attributes * @return */ public Integer getCategory(Object[] attributes) { return m_category; } /** * Tworzy bardziej przejrzystą strukturę drzewa * @return */ public ShowedNode showTree() { ShowedNode node = new ShowedNode(); node.P = new Integer[m_P.size()]; m_P.toArray(node.P); node.Category = this.category(); return node; } }