如何在teradataml中使用NaiveBayes建立模型？

Question

如何在Teradataml中使用NaiveBayes构建模型: 1.输入数据为teradata表

Answer 1

1. 导入如下：

从teradataml.context.context导入teradataml导入*从teradataml.dataframe.dataframe导入DataFrame从teradataml.analytics.mle.NaiveBayesPredict导入teradataml用于连接到所需teradata数据库的数据库上下文

con =create_context(host=“主机名”，username=“指定用户名”，password=“指定密码”)

2、为驻留在teradata中的输入表创建数据帧

nb_input_train = DataFrame.from_table("nb_input_train")
nb_input_test = DataFrame.from_table("nb_input_test")

1. 列车: naivebayes_train = NaiveBayes(formula="col1 ~ col2 + col3 + col4 + col5"，data=nb_input_train)
2. Predict

naivebayes_predict_result = NaiveBayesPredict(newdata=nb_input_test，modeldata=naivebayes_train，id_col="id"，formula="col1 ~ col2 + col3 + col4 + col5"，responses = "val1"，"val2"，"val3")

提示：为了查看sqlmr查询，将print_sqlmr_query设置为true

from teradataml.options.display import display
display.print_sqlmr_query = True