Tutorials Code: Source Code for Tutorial 4

From DSL

Jump to: navigation, search
  #include "smilearn.h"
  #include "smile.h"
  #include <iostream>
  
  using namespace std;
  
  void NaiveBayes()
  {
  DSL_textParser parser;
  parser.SetUseHeader(true);
  
  if (parser.Parse("naive.txt")!=DSL_OKAY)
  cout << "Parsing failed!" << endl;
  DSL_dataset d = parser.GetDataset();
  
  DSL_network result;
  DSL_naiveBayes naive;
  naive.classVariableId = "c";
  if (naive.Learn(d,result)!=DSL_OKAY)
  {
  cout << "Learning failed." << endl;
  return;
  }
  
  result.WriteFile("naivebayes.xdsl");
  }
  
  
  int main()
  {
  NaiveBayes();
  }
  
  /*
  c x y
  State1 State1 State0
  State1 State1 State0
  State0 State1 State1
  State0 State0 State0
  State0 State0 State1
  State0 State0 State1
  State0 State1 State0
  State0 State0 State1
  State0 State1 State1
  State0 State1 State1
  */
Personal tools