【发布时间】:2019-12-26 06:40:20
【问题描述】:
我正在尝试实现我的文本分类模型,该模型使用 SVM 模型在 Tensorflow 中将消息分类为垃圾邮件或非垃圾邮件。 我只有 2 个特征,即联系人姓名和消息,我需要将它们转换为数字还是可以直接用作特征列中的输入
svm = tf.contrib.learn.SVM(
example_id_column=example_id_column_name,
feature_columns=(tf.contrib.layers.real_valued_column(
column_name=x_column_name, dimension=90),),
l2_regularization=0.1)
svm.fit(input_fn=train_input_fn, steps=10)
【问题讨论】:
标签: tensorflow machine-learning svm text-classification