Question

我的数据库中有大约1600篇文章，每篇文章都预先标有以下类别之一：

Technology
Science
Business
World
Health
Entertainment
Sports

我正在尝试使用sci-kit学习构建一个分类新文章的分类器。（我想我会将训练数据分成两半，用于训练和测试？）

我希望使用tf-idf，因为我没有停止词的列表（我可以使用NLTK只提取形容词和名词，但我宁愿给scikit - 学习完整的文章）。

我已经阅读了关于scikit-learn的所有文档，但是他们的例子涉及单词出现和N-gram（很好），但他们从未指定如何将一段数据绑定到标签。

我已经尝试过查看他们的示例代码了，但是这样做太难以理解了。

有人可以帮我解决这个问题，还是指出我正确的方向？

感谢。

Answer 1

我认为你在我开始将自己的数据提供给分类器时遇到了同样的问题。

您可以使用函数sklearn.datasets.load_files，但为此，您需要创建此结构：

train
├── science
│   ├── 0001.txt
│   └── 0002.txt
└── technology
    ├── 0001.txt
    └── 0002.txt

train的子目录被命名为标签，而labels目录中的每个文件都是具有该对应标签的文章。然后使用load_files加载数据：

In [1]: from sklearn.datasets import load_files

In [2]: load_files('train')
Out[2]: 
{'DESCR': None,
 'data': ['iphone apple smartphone\n',
  'linux windows ubuntu\n',
  'biology astrophysics\n',
  'math\n'],
 'filenames': array(['train/technology/0001.txt', 'train/technology/0002.txt',
       'train/science/0002.txt', 'train/science/0001.txt'], 
      dtype='|S25'),
 'target': array([1, 1, 0, 0]),
 'target_names': ['science', 'technology']}

返回的对象是sklearn.datasets.base.Bunch，它是一个简单的数据包装器。这是一种开始使用分类器的直接方法，但是当您的数据更大并且频繁更改时，您可能希望停止使用文件并使用例如数据库来存储标记文档，并且可能具有比仅仅普通文件更多的结构文本。基本上，您需要生成类别列表（或target_names），例如['science', 'technology', ...]，并为target列表中的每个文档指定data值作为标记的索引target_names列表中的类别。 data和target的长度必须相同。

您可以查看我之前写的这个脚本来运行分类器：https://github.com/darkrho/yatiri/blob/master/scripts/run_classifier.py#L267

Answer 2

也许从这里的示例开始：http://scikit-learn.org/dev/auto_examples/document_classification_20newsgroups.html#example-document-classification-20newsgroups-py

这是一个更高级的例子：http://scikit-learn.org/dev/auto_examples/grid_search_text_feature_extraction.html#example-grid-search-text-feature-extraction-py

示例库中还有更多文本示例： http://scikit-learn.org/dev/auto_examples/index.html

使用Sci-Kit学习使用大型语料库对文本进行分类

2 个答案: