Using the JDM API - Java Data Mining: Strategy, Standard, and Practice - page 201

4. import javax.datamining.resource.*;
5. /** This class build a classification model with default settings */
6. public class CustomerAttrition {
7. //Data members
8. static ConnectionFactory connFactory = null;
9. static Connection dmeConn = null;
10. static PhysicalDataSetFactory pdsFactory = null;
11. static PhysicalAttributeFactory paFactory = null;
12. static ClassificationSettingsFactory clsFactory = null;
13. static BuildTaskFactory btkFactory = null;
14. //Main method
15. public static void main(String args[]) throws Exception {
16. CustomerAttrition attrition_model = new CustomerAttrition();
17. try { attrition_model.reset() } catch(Exception anyCleanupFailures) {}//Ignore
18. try {
19. attrition_model.init(); //Create object factories
20. attrition_model.input(); //Create input objects
21. if( attrition_model.run() ) //Execute model build task
22. attrition_model.output(); //Retrieve the model
23. } catch(Exception anyFailures) { anyFailures.printStackTrace();
24. } finally { if(dmeConn != null) dmeConn.close(); //Close connection
25. }
26. }
27. //Create object factories using the DME connection
28. public void init() throws JDMException {
29. //Create DME connection as shown in Listing 9.1 or 9.2
30. // . . .
31. //Create object factories for the objects in this example
32. pdsFactory = (PhysicalDataSetFactory)dmeConn.getFactory(
33. "javax.datamining.data.PhysicalDataSet");
34. clsFactory = (ClassificationSettingsFactory)dmeConn.getFactory(
35. "javax.datamining.supervised.classification.ClassificationSettings");
36. btkFactory = (BuildTaskFactory)dmeConn.getFactory(
37. "javax.datamining.task.BuildTask");
38. }
39. //Create and save input objects for build task
40. public void input() throws JDMException {
41. PhysicalDataSet pds = pdsFactory.create( "CUSTOMERS_BUILD_DATA", false );
42. ClassificationSettings cls = clsFactory.create();
43. cls.setTargetAttributeName("Attrition");
44. boolean replaceExistingObject = true;
45. dmeConn.saveObject( "attrition_build_data", pds, replaceExistingObject );
46. dmeConn.saveObject( "attrition_build_settings", cls, replaceExistingObject );
47. }
48. //Create build task and execute it to build model
49. public boolean run() throws JDMException {
50. BuildTask btk = btkFactory.create( "attrition_build_data",
51. "attrition_build_settings", "attrition_model" );
52. Long timeOut=null;//Run until completion