package MachineLearning; import java.util.*; public class AttrChangeCollection { private HashMap m_attrChangeCollection; public AttrChangeCollection() { m_attrChangeCollection = new HashMap(); } public void Add(ArrayList tresholds, Integer attrIndex) { if (!m_attrChangeCollection.containsKey(attrIndex)) { AttributChange attrChange = new AttributChange(tresholds,attrIndex); m_attrChangeCollection.put(attrIndex, attrChange); } } public void Change(Object[][] examples, Integer attrIndex) { if(m_attrChangeCollection.containsKey(attrIndex)) { m_attrChangeCollection.get(attrIndex).Change(examples); } } public void Change(Object[][] examples) { for (AttributChange attrChange : m_attrChangeCollection.values()) { attrChange.Change(examples); } } public void Change(Object[] example) { for (AttributChange attrChange : m_attrChangeCollection.values()) { attrChange.Change(example); } } public void WriteAggregateValues() { if (m_attrChangeCollection.size() > 0) { WriteFile sw = new WriteFile(); sw.open("Aggregate.txt"); sw.writeLine("Nowy atrybut"); sw.writeLine("\tStary atrybut"); m_attrChangeCollection.get(0).WriteOldNewValues("DaysOfWeek", sw, Days.values()); m_attrChangeCollection.get(1).WriteOldNewValues("Hour", sw); m_attrChangeCollection.get(2).WriteOldNewValues("Month", sw, Month.values()); sw.close(); } } }