Unity 2018 Artificial Intelligence Cookbook - Second Edition

How to do it...

The classifier we'll build won't take long, just five steps:

1. Create the class and its member variables:

using UnityEngine; 
using System.Collections; 
using System.Collections.Generic; 
 
public class NaiveBayesClassifier : MonoBehaviour 
{ 
    public int numAttributes; 
    public int numExamplesPositive; 
    public int numExamplesNegative; 
 
    public List<bool> attrCountPositive; 
    public List<bool> attrCountNegative; 
} 

2. Define the Awake method for initialization:

void Awake() 
{ 
    attrCountPositive = new List<bool>(); 
    attrCountNegative = new List<bool>(); 
} 

3. Implement the function for updating the classifier:

public void UpdateClassifier(bool[] attributes, NBCLabel label) { if (label == NBCLabel.POSITIVE) { numExamplesPositive++; attrCountPositive.AddRange(attributes); ...