gaussianNB
语法
gaussianNB(Y, X, [varSmoothing=1e-9])
参数
Y 是一个长度与X的行数相等的向量,表示X中每个样本对应的标签。
X 是一个表,表示训练集。表中的每一行表示一个样本,每一列表示一个特征。
varSmoothing 是一个浮点类型的正数,表示平滑系数。默认值是1e-9。
详情
使用高斯朴素贝叶斯(Gaussian Naive Bayes)算法对数据进行分类训练。返回的结果是一个字典,包含以下key:
modelName:模型名称,为字符串”GaussianNB”
model:gaussianNB的内部模型
varSmoothing:训练时的平滑系数
例子
本例所用数据集iris.data可从 https://archive.ics.uci.edu/ml/datasets/iris 下载。
$ DATA_DIR = "C:/DolphinDB/Data"
$ t = loadText(DATA_DIR+"/iris.data")
$ t.rename!(`col0`col1`col2`col3`col4, `sepalLength`sepalWidth`petalLength`petalWidth`class)
$ t[`classType] = take(0, t.size())
$ update t set classType = 1 where class = "Iris-versicolor"
$ update t set classType = 2 where class = "Iris-virginica"
$ training = select sepalLength, sepalWidth, petalLength, petalWidth from t
$ labels = t.classType
$ model = gaussianNB(labels, training);
$ predict(model, training);