/doc/tagger.build_dict-pysrc.html
HTML | 250 lines | 219 code | 10 blank | 21 comment | 0 complexity | a85fec16986c0c11d83ce7676af49e6d MD5 | raw file
- <?xml version="1.0" encoding="ascii"?>
- <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
- "DTD/xhtml1-transitional.dtd">
- <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
- <head>
- <title>tagger.build_dict</title>
- <link rel="stylesheet" href="epydoc.css" type="text/css" />
- <script type="text/javascript" src="epydoc.js"></script>
- </head>
- <body bgcolor="white" text="black" link="blue" vlink="#204080"
- alink="#204080">
- <!-- ==================== NAVIGATION BAR ==================== -->
- <table class="navbar" border="0" width="100%" cellpadding="0"
- bgcolor="#a0c0ff" cellspacing="0">
- <tr valign="middle">
- <!-- Tree link -->
- <th> <a
- href="module-tree.html">Trees</a> </th>
- <!-- Index link -->
- <th> <a
- href="identifier-index.html">Indices</a> </th>
- <!-- Help link -->
- <th> <a
- href="help.html">Help</a> </th>
- <!-- Project homepage -->
- <th class="navbar" align="right" width="100%">
- <table border="0" cellpadding="0" cellspacing="0">
- <tr><th class="navbar" align="center"
- ><a class="navbar" target="_top" href="http://github.com/apresta/tagger">tagger</a></th>
- </tr></table></th>
- </tr>
- </table>
- <table width="100%" cellpadding="0" cellspacing="0">
- <tr valign="top">
- <td width="100%">
- <span class="breadcrumbs">
- Package tagger ::
- Module build_dict
- </span>
- </td>
- <td>
- <table cellpadding="0" cellspacing="0">
- <!-- hide/show private -->
- <tr><td align="right"><span class="options">[<a href="javascript:void(0);" class="privatelink"
- onclick="toggle_private();">hide private</a>]</span></td></tr>
- <tr><td align="right"><span class="options"
- >[<a href="frames.html" target="_top">frames</a
- >] | <a href="tagger.build_dict-pysrc.html"
- target="_top">no frames</a>]</span></td></tr>
- </table>
- </td>
- </tr>
- </table>
- <h1 class="epydoc">Source Code for <a href="tagger.build_dict-module.html">Module tagger.build_dict</a></h1>
- <pre class="py-src">
- <a name="L1"></a><tt class="py-lineno"> 1</tt> <tt class="py-line"><tt class="py-comment">#!/usr/bin/env python</tt> </tt>
- <a name="L2"></a><tt class="py-lineno"> 2</tt> <tt class="py-line"> </tt>
- <a name="L3"></a><tt class="py-lineno"> 3</tt> <tt class="py-line"><tt class="py-comment"># Copyright (C) 2011 by Alessandro Presta</tt> </tt>
- <a name="L4"></a><tt class="py-lineno"> 4</tt> <tt class="py-line"> </tt>
- <a name="L5"></a><tt class="py-lineno"> 5</tt> <tt class="py-line"><tt class="py-comment"># Permission is hereby granted, free of charge, to any person obtaining a copy</tt> </tt>
- <a name="L6"></a><tt class="py-lineno"> 6</tt> <tt class="py-line"><tt class="py-comment"># of this software and associated documentation files (the "Software"), to deal</tt> </tt>
- <a name="L7"></a><tt class="py-lineno"> 7</tt> <tt class="py-line"><tt class="py-comment"># in the Software without restriction, including without limitation the rights</tt> </tt>
- <a name="L8"></a><tt class="py-lineno"> 8</tt> <tt class="py-line"><tt class="py-comment"># to use, copy, modify, merge, publish, distribute, sublicense, and/or sell</tt> </tt>
- <a name="L9"></a><tt class="py-lineno"> 9</tt> <tt class="py-line"><tt class="py-comment"># copies of the Software, and to permit persons to whom the Software is</tt> </tt>
- <a name="L10"></a><tt class="py-lineno"> 10</tt> <tt class="py-line"><tt class="py-comment"># furnished to do so, subject to the following conditions:</tt> </tt>
- <a name="L11"></a><tt class="py-lineno"> 11</tt> <tt class="py-line"> </tt>
- <a name="L12"></a><tt class="py-lineno"> 12</tt> <tt class="py-line"><tt class="py-comment"># The above copyright notice and this permission notice shall be included in</tt> </tt>
- <a name="L13"></a><tt class="py-lineno"> 13</tt> <tt class="py-line"><tt class="py-comment"># all copies or substantial portions of the Software.</tt> </tt>
- <a name="L14"></a><tt class="py-lineno"> 14</tt> <tt class="py-line"> </tt>
- <a name="L15"></a><tt class="py-lineno"> 15</tt> <tt class="py-line"><tt class="py-comment"># THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR</tt> </tt>
- <a name="L16"></a><tt class="py-lineno"> 16</tt> <tt class="py-line"><tt class="py-comment"># IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,</tt> </tt>
- <a name="L17"></a><tt class="py-lineno"> 17</tt> <tt class="py-line"><tt class="py-comment"># FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE</tt> </tt>
- <a name="L18"></a><tt class="py-lineno"> 18</tt> <tt class="py-line"><tt class="py-comment"># AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER</tt> </tt>
- <a name="L19"></a><tt class="py-lineno"> 19</tt> <tt class="py-line"><tt class="py-comment"># LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,</tt> </tt>
- <a name="L20"></a><tt class="py-lineno"> 20</tt> <tt class="py-line"><tt class="py-comment"># OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN</tt> </tt>
- <a name="L21"></a><tt class="py-lineno"> 21</tt> <tt class="py-line"><tt class="py-comment"># THE SOFTWARE</tt> </tt>
- <a name="L22"></a><tt class="py-lineno"> 22</tt> <tt class="py-line"> </tt>
- <a name="L23"></a><tt class="py-lineno"> 23</tt> <tt class="py-line"> </tt>
- <a name="L24"></a><tt class="py-lineno"> 24</tt> <tt class="py-line"><tt class="py-docstring">'''</tt> </tt>
- <a name="L25"></a><tt class="py-lineno"> 25</tt> <tt class="py-line"><tt class="py-docstring">Usage: build_dict.py -o <output file> -s <stopwords file> <list of files></tt> </tt>
- <a name="L26"></a><tt class="py-lineno"> 26</tt> <tt class="py-line"><tt class="py-docstring">'''</tt> </tt>
- <a name="L27"></a><tt class="py-lineno"> 27</tt> <tt class="py-line"> </tt>
- <a name="L28"></a><tt class="py-lineno"> 28</tt> <tt class="py-line"><tt class="py-keyword">from</tt> <tt class="py-name">__future__</tt> <tt class="py-keyword">import</tt> <tt class="py-name">division</tt> </tt>
- <a name="L29"></a><tt class="py-lineno"> 29</tt> <tt class="py-line"> </tt>
- <a name="L30"></a><tt class="py-lineno"> 30</tt> <tt class="py-line"><tt class="py-keyword">from</tt> <tt id="link-0" class="py-name" targets="Module tagger.tagger=tagger.tagger-module.html"><a title="tagger.tagger" class="py-name" href="#" onclick="return doclink('link-0', 'tagger', 'link-0');">tagger</a></tt> <tt class="py-keyword">import</tt> <tt id="link-1" class="py-name" targets="Class tagger.tagger.Stemmer=tagger.tagger.Stemmer-class.html"><a title="tagger.tagger.Stemmer" class="py-name" href="#" onclick="return doclink('link-1', 'Stemmer', 'link-1');">Stemmer</a></tt> </tt>
- <a name="L31"></a><tt class="py-lineno"> 31</tt> <tt class="py-line"><tt class="py-keyword">from</tt> <tt id="link-2" class="py-name" targets="Module tagger.extras=tagger.extras-module.html"><a title="tagger.extras" class="py-name" href="#" onclick="return doclink('link-2', 'extras', 'link-2');">extras</a></tt> <tt class="py-keyword">import</tt> <tt id="link-3" class="py-name" targets="Class tagger.extras.SimpleReader=tagger.extras.SimpleReader-class.html"><a title="tagger.extras.SimpleReader" class="py-name" href="#" onclick="return doclink('link-3', 'SimpleReader', 'link-3');">SimpleReader</a></tt> </tt>
- <a name="L32"></a><tt class="py-lineno"> 32</tt> <tt class="py-line"> </tt>
- <a name="build_dict"></a><div id="build_dict-def"><a name="L33"></a><tt class="py-lineno"> 33</tt> <a class="py-toggle" href="#" id="build_dict-toggle" onclick="return toggle('build_dict');">-</a><tt class="py-line"><tt class="py-keyword">def</tt> <a class="py-def-name" href="tagger.build_dict-module.html#build_dict">build_dict</a><tt class="py-op">(</tt><tt class="py-param">corpus</tt><tt class="py-op">,</tt> <tt class="py-param">stopwords</tt><tt class="py-op">=</tt><tt class="py-name">None</tt><tt class="py-op">,</tt> <tt class="py-param">measure</tt><tt class="py-op">=</tt><tt class="py-string">'IDF'</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
- </div><div id="build_dict-collapsed" style="display:none;" pad="+++" indent="++++"></div><div id="build_dict-expanded"><a name="L34"></a><tt class="py-lineno"> 34</tt> <tt class="py-line"> <tt class="py-docstring">'''</tt> </tt>
- <a name="L35"></a><tt class="py-lineno"> 35</tt> <tt class="py-line"><tt class="py-docstring"> @param corpus: a list of documents, represented as lists of (stemmed)</tt> </tt>
- <a name="L36"></a><tt class="py-lineno"> 36</tt> <tt class="py-line"><tt class="py-docstring"> words</tt> </tt>
- <a name="L37"></a><tt class="py-lineno"> 37</tt> <tt class="py-line"><tt class="py-docstring"> @param stopwords: the list of (stemmed) words that should have zero weight</tt> </tt>
- <a name="L38"></a><tt class="py-lineno"> 38</tt> <tt class="py-line"><tt class="py-docstring"> @param measure: the measure used to compute the weights ('IDF'</tt> </tt>
- <a name="L39"></a><tt class="py-lineno"> 39</tt> <tt class="py-line"><tt class="py-docstring"> i.e. 'inverse document frequency' or 'ICF' i.e.</tt> </tt>
- <a name="L40"></a><tt class="py-lineno"> 40</tt> <tt class="py-line"><tt class="py-docstring"> 'inverse collection frequency'; defaults to 'IDF')</tt> </tt>
- <a name="L41"></a><tt class="py-lineno"> 41</tt> <tt class="py-line"><tt class="py-docstring"></tt> </tt>
- <a name="L42"></a><tt class="py-lineno"> 42</tt> <tt class="py-line"><tt class="py-docstring"> @returns: a dictionary of weights in the interval [0,1]</tt> </tt>
- <a name="L43"></a><tt class="py-lineno"> 43</tt> <tt class="py-line"><tt class="py-docstring"> '''</tt> </tt>
- <a name="L44"></a><tt class="py-lineno"> 44</tt> <tt class="py-line"> </tt>
- <a name="L45"></a><tt class="py-lineno"> 45</tt> <tt class="py-line"> <tt class="py-keyword">import</tt> <tt class="py-name">collections</tt> </tt>
- <a name="L46"></a><tt class="py-lineno"> 46</tt> <tt class="py-line"> <tt class="py-keyword">import</tt> <tt class="py-name">math</tt> </tt>
- <a name="L47"></a><tt class="py-lineno"> 47</tt> <tt class="py-line"> </tt>
- <a name="L48"></a><tt class="py-lineno"> 48</tt> <tt class="py-line"> <tt class="py-name">dictionary</tt> <tt class="py-op">=</tt> <tt class="py-op">{</tt><tt class="py-op">}</tt> </tt>
- <a name="L49"></a><tt class="py-lineno"> 49</tt> <tt class="py-line"> </tt>
- <a name="L50"></a><tt class="py-lineno"> 50</tt> <tt class="py-line"> <tt class="py-keyword">if</tt> <tt class="py-name">measure</tt> <tt class="py-op">==</tt> <tt class="py-string">'ICF'</tt><tt class="py-op">:</tt> </tt>
- <a name="L51"></a><tt class="py-lineno"> 51</tt> <tt class="py-line"> <tt class="py-name">words</tt> <tt class="py-op">=</tt> <tt class="py-op">[</tt><tt class="py-name">w</tt> <tt class="py-keyword">for</tt> <tt class="py-name">doc</tt> <tt class="py-keyword">in</tt> <tt class="py-name">corpus</tt> <tt class="py-keyword">for</tt> <tt class="py-name">w</tt> <tt class="py-keyword">in</tt> <tt class="py-name">doc</tt><tt class="py-op">]</tt> </tt>
- <a name="L52"></a><tt class="py-lineno"> 52</tt> <tt class="py-line"> </tt>
- <a name="L53"></a><tt class="py-lineno"> 53</tt> <tt class="py-line"> <tt class="py-name">term_count</tt> <tt class="py-op">=</tt> <tt class="py-name">collections</tt><tt class="py-op">.</tt><tt class="py-name">Counter</tt><tt class="py-op">(</tt><tt class="py-name">words</tt><tt class="py-op">)</tt> </tt>
- <a name="L54"></a><tt class="py-lineno"> 54</tt> <tt class="py-line"> <tt class="py-name">total_count</tt> <tt class="py-op">=</tt> <tt class="py-name">len</tt><tt class="py-op">(</tt><tt class="py-name">words</tt><tt class="py-op">)</tt> </tt>
- <a name="L55"></a><tt class="py-lineno"> 55</tt> <tt class="py-line"> <tt class="py-name">scale</tt> <tt class="py-op">=</tt> <tt class="py-name">math</tt><tt class="py-op">.</tt><tt class="py-name">log</tt><tt class="py-op">(</tt><tt class="py-name">total_count</tt><tt class="py-op">)</tt> </tt>
- <a name="L56"></a><tt class="py-lineno"> 56</tt> <tt class="py-line"> </tt>
- <a name="L57"></a><tt class="py-lineno"> 57</tt> <tt class="py-line"> <tt class="py-keyword">for</tt> <tt class="py-name">w</tt><tt class="py-op">,</tt> <tt class="py-name">cnt</tt> <tt class="py-keyword">in</tt> <tt class="py-name">term_count</tt><tt class="py-op">.</tt><tt class="py-name">iteritems</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
- <a name="L58"></a><tt class="py-lineno"> 58</tt> <tt class="py-line"> <tt class="py-name">dictionary</tt><tt class="py-op">[</tt><tt class="py-name">w</tt><tt class="py-op">]</tt> <tt class="py-op">=</tt> <tt class="py-name">math</tt><tt class="py-op">.</tt><tt class="py-name">log</tt><tt class="py-op">(</tt><tt class="py-name">total_count</tt> <tt class="py-op">/</tt> <tt class="py-op">(</tt><tt class="py-name">cnt</tt> <tt class="py-op">+</tt> <tt class="py-number">1</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> <tt class="py-op">/</tt> <tt class="py-name">scale</tt> </tt>
- <a name="L59"></a><tt class="py-lineno"> 59</tt> <tt class="py-line"> </tt>
- <a name="L60"></a><tt class="py-lineno"> 60</tt> <tt class="py-line"> <tt class="py-keyword">elif</tt> <tt class="py-name">measure</tt> <tt class="py-op">==</tt> <tt class="py-string">'IDF'</tt><tt class="py-op">:</tt> </tt>
- <a name="L61"></a><tt class="py-lineno"> 61</tt> <tt class="py-line"> <tt class="py-name">corpus_size</tt> <tt class="py-op">=</tt> <tt class="py-name">len</tt><tt class="py-op">(</tt><tt class="py-name">corpus</tt><tt class="py-op">)</tt> </tt>
- <a name="L62"></a><tt class="py-lineno"> 62</tt> <tt class="py-line"> <tt class="py-name">scale</tt> <tt class="py-op">=</tt> <tt class="py-name">math</tt><tt class="py-op">.</tt><tt class="py-name">log</tt><tt class="py-op">(</tt><tt class="py-name">corpus_size</tt><tt class="py-op">)</tt> </tt>
- <a name="L63"></a><tt class="py-lineno"> 63</tt> <tt class="py-line"> </tt>
- <a name="L64"></a><tt class="py-lineno"> 64</tt> <tt class="py-line"> <tt class="py-name">term_count</tt> <tt class="py-op">=</tt> <tt class="py-name">collections</tt><tt class="py-op">.</tt><tt class="py-name">defaultdict</tt><tt class="py-op">(</tt><tt class="py-name">int</tt><tt class="py-op">)</tt> </tt>
- <a name="L65"></a><tt class="py-lineno"> 65</tt> <tt class="py-line"> </tt>
- <a name="L66"></a><tt class="py-lineno"> 66</tt> <tt class="py-line"> <tt class="py-keyword">for</tt> <tt class="py-name">doc</tt> <tt class="py-keyword">in</tt> <tt class="py-name">corpus</tt><tt class="py-op">:</tt> </tt>
- <a name="L67"></a><tt class="py-lineno"> 67</tt> <tt class="py-line"> <tt class="py-name">words</tt> <tt class="py-op">=</tt> <tt class="py-name">set</tt><tt class="py-op">(</tt><tt class="py-name">doc</tt><tt class="py-op">)</tt> </tt>
- <a name="L68"></a><tt class="py-lineno"> 68</tt> <tt class="py-line"> <tt class="py-keyword">for</tt> <tt class="py-name">w</tt> <tt class="py-keyword">in</tt> <tt class="py-name">words</tt><tt class="py-op">:</tt> </tt>
- <a name="L69"></a><tt class="py-lineno"> 69</tt> <tt class="py-line"> <tt class="py-name">term_count</tt><tt class="py-op">[</tt><tt class="py-name">w</tt><tt class="py-op">]</tt> <tt class="py-op">+=</tt> <tt class="py-number">1</tt> </tt>
- <a name="L70"></a><tt class="py-lineno"> 70</tt> <tt class="py-line"> </tt>
- <a name="L71"></a><tt class="py-lineno"> 71</tt> <tt class="py-line"> <tt class="py-keyword">for</tt> <tt class="py-name">w</tt><tt class="py-op">,</tt> <tt class="py-name">cnt</tt> <tt class="py-keyword">in</tt> <tt class="py-name">term_count</tt><tt class="py-op">.</tt><tt class="py-name">iteritems</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
- <a name="L72"></a><tt class="py-lineno"> 72</tt> <tt class="py-line"> <tt class="py-name">dictionary</tt><tt class="py-op">[</tt><tt class="py-name">w</tt><tt class="py-op">]</tt> <tt class="py-op">=</tt> <tt class="py-name">math</tt><tt class="py-op">.</tt><tt class="py-name">log</tt><tt class="py-op">(</tt><tt class="py-name">corpus_size</tt> <tt class="py-op">/</tt> <tt class="py-op">(</tt><tt class="py-name">cnt</tt> <tt class="py-op">+</tt> <tt class="py-number">1</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> <tt class="py-op">/</tt> <tt class="py-name">scale</tt> </tt>
- <a name="L73"></a><tt class="py-lineno"> 73</tt> <tt class="py-line"> </tt>
- <a name="L74"></a><tt class="py-lineno"> 74</tt> <tt class="py-line"> <tt class="py-keyword">if</tt> <tt class="py-name">stopwords</tt><tt class="py-op">:</tt> </tt>
- <a name="L75"></a><tt class="py-lineno"> 75</tt> <tt class="py-line"> <tt class="py-keyword">for</tt> <tt class="py-name">w</tt> <tt class="py-keyword">in</tt> <tt class="py-name">stopwords</tt><tt class="py-op">:</tt> </tt>
- <a name="L76"></a><tt class="py-lineno"> 76</tt> <tt class="py-line"> <tt class="py-name">dictionary</tt><tt class="py-op">[</tt><tt class="py-name">w</tt><tt class="py-op">]</tt> <tt class="py-op">=</tt> <tt class="py-number">0.0</tt> </tt>
- <a name="L77"></a><tt class="py-lineno"> 77</tt> <tt class="py-line"> </tt>
- <a name="L78"></a><tt class="py-lineno"> 78</tt> <tt class="py-line"> <tt class="py-keyword">return</tt> <tt class="py-name">dictionary</tt> </tt>
- </div><a name="L79"></a><tt class="py-lineno"> 79</tt> <tt class="py-line"> </tt>
- <a name="L80"></a><tt class="py-lineno"> 80</tt> <tt class="py-line"> </tt>
- <a name="build_dict_from_files"></a><div id="build_dict_from_files-def"><a name="L81"></a><tt class="py-lineno"> 81</tt> <a class="py-toggle" href="#" id="build_dict_from_files-toggle" onclick="return toggle('build_dict_from_files');">-</a><tt class="py-line"><tt class="py-keyword">def</tt> <a class="py-def-name" href="tagger.build_dict-module.html#build_dict_from_files">build_dict_from_files</a><tt class="py-op">(</tt><tt class="py-param">output_file</tt><tt class="py-op">,</tt> <tt class="py-param">corpus_files</tt><tt class="py-op">,</tt> <tt class="py-param">stopwords_file</tt><tt class="py-op">=</tt><tt class="py-name">None</tt><tt class="py-op">,</tt> </tt>
- <a name="L82"></a><tt class="py-lineno"> 82</tt> <tt class="py-line"> <tt class="py-param">reader</tt><tt class="py-op">=</tt><tt id="link-4" class="py-name"><a title="tagger.extras.SimpleReader" class="py-name" href="#" onclick="return doclink('link-4', 'SimpleReader', 'link-3');">SimpleReader</a></tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">,</tt> <tt class="py-param">stemmer</tt><tt class="py-op">=</tt><tt id="link-5" class="py-name"><a title="tagger.tagger.Stemmer" class="py-name" href="#" onclick="return doclink('link-5', 'Stemmer', 'link-1');">Stemmer</a></tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">,</tt> </tt>
- <a name="L83"></a><tt class="py-lineno"> 83</tt> <tt class="py-line"> <tt class="py-param">measure</tt><tt class="py-op">=</tt><tt class="py-string">'IDF'</tt><tt class="py-op">,</tt> <tt class="py-param">verbose</tt><tt class="py-op">=</tt><tt class="py-name">False</tt><tt class="py-op">)</tt><tt class="py-op">:</tt> </tt>
- </div><div id="build_dict_from_files-collapsed" style="display:none;" pad="+++" indent="++++"></div><div id="build_dict_from_files-expanded"><a name="L84"></a><tt class="py-lineno"> 84</tt> <tt class="py-line"> <tt class="py-docstring">'''</tt> </tt>
- <a name="L85"></a><tt class="py-lineno"> 85</tt> <tt class="py-line"><tt class="py-docstring"> @param output_file: the name of the file where the dictionary should be</tt> </tt>
- <a name="L86"></a><tt class="py-lineno"> 86</tt> <tt class="py-line"><tt class="py-docstring"> saved</tt> </tt>
- <a name="L87"></a><tt class="py-lineno"> 87</tt> <tt class="py-line"><tt class="py-docstring"> @param corpus_files: a list of files with words to process</tt> </tt>
- <a name="L88"></a><tt class="py-lineno"> 88</tt> <tt class="py-line"><tt class="py-docstring"> @param stopwords_file: a file containing a list of stopwords</tt> </tt>
- <a name="L89"></a><tt class="py-lineno"> 89</tt> <tt class="py-line"><tt class="py-docstring"> @param reader: the L{Reader} object to be used</tt> </tt>
- <a name="L90"></a><tt class="py-lineno"> 90</tt> <tt class="py-line"><tt class="py-docstring"> @param stemmer: the L{Stemmer} object to be used</tt> </tt>
- <a name="L91"></a><tt class="py-lineno"> 91</tt> <tt class="py-line"><tt class="py-docstring"> @param measure: the measure used to compute the weights ('IDF'</tt> </tt>
- <a name="L92"></a><tt class="py-lineno"> 92</tt> <tt class="py-line"><tt class="py-docstring"> i.e. 'inverse document frequency' or 'ICF' i.e.</tt> </tt>
- <a name="L93"></a><tt class="py-lineno"> 93</tt> <tt class="py-line"><tt class="py-docstring"> 'inverse collection frequency'; defaults to 'IDF')</tt> </tt>
- <a name="L94"></a><tt class="py-lineno"> 94</tt> <tt class="py-line"><tt class="py-docstring"> @param verbose: whether information on the progress should be</tt> </tt>
- <a name="L95"></a><tt class="py-lineno"> 95</tt> <tt class="py-line"><tt class="py-docstring"> printed on screen</tt> </tt>
- <a name="L96"></a><tt class="py-lineno"> 96</tt> <tt class="py-line"><tt class="py-docstring"> '''</tt> </tt>
- <a name="L97"></a><tt class="py-lineno"> 97</tt> <tt class="py-line"> </tt>
- <a name="L98"></a><tt class="py-lineno"> 98</tt> <tt class="py-line"> <tt class="py-keyword">import</tt> <tt class="py-name">pickle</tt> </tt>
- <a name="L99"></a><tt class="py-lineno"> 99</tt> <tt class="py-line"> </tt>
- <a name="L100"></a><tt class="py-lineno">100</tt> <tt class="py-line"> <tt class="py-keyword">if</tt> <tt class="py-name">verbose</tt><tt class="py-op">:</tt> <tt class="py-keyword">print</tt> <tt class="py-string">'Processing corpus...'</tt> </tt>
- <a name="L101"></a><tt class="py-lineno">101</tt> <tt class="py-line"> <tt class="py-name">corpus</tt> <tt class="py-op">=</tt> <tt class="py-op">[</tt><tt class="py-op">]</tt> </tt>
- <a name="L102"></a><tt class="py-lineno">102</tt> <tt class="py-line"> <tt class="py-keyword">for</tt> <tt class="py-name">filename</tt> <tt class="py-keyword">in</tt> <tt class="py-name">corpus_files</tt><tt class="py-op">:</tt> </tt>
- <a name="L103"></a><tt class="py-lineno">103</tt> <tt class="py-line"> <tt class="py-keyword">with</tt> <tt class="py-name">open</tt><tt class="py-op">(</tt><tt class="py-name">filename</tt><tt class="py-op">,</tt> <tt class="py-string">'r'</tt><tt class="py-op">)</tt> <tt class="py-keyword">as</tt> <tt class="py-name">doc</tt><tt class="py-op">:</tt> </tt>
- <a name="L104"></a><tt class="py-lineno">104</tt> <tt class="py-line"> <tt class="py-name">corpus</tt><tt class="py-op">.</tt><tt class="py-name">append</tt><tt class="py-op">(</tt><tt class="py-name">reader</tt><tt class="py-op">(</tt><tt class="py-name">doc</tt><tt class="py-op">.</tt><tt class="py-name">read</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
- <a name="L105"></a><tt class="py-lineno">105</tt> <tt class="py-line"> <tt class="py-name">corpus</tt> <tt class="py-op">=</tt> <tt class="py-op">[</tt><tt class="py-op">[</tt><tt class="py-name">w</tt><tt class="py-op">.</tt><tt class="py-name">stem</tt> <tt class="py-keyword">for</tt> <tt class="py-name">w</tt> <tt class="py-keyword">in</tt> <tt class="py-name">map</tt><tt class="py-op">(</tt><tt class="py-name">stemmer</tt><tt class="py-op">,</tt> <tt class="py-name">doc</tt><tt class="py-op">)</tt><tt class="py-op">]</tt> <tt class="py-keyword">for</tt> <tt class="py-name">doc</tt> <tt class="py-keyword">in</tt> <tt class="py-name">corpus</tt><tt class="py-op">]</tt> </tt>
- <a name="L106"></a><tt class="py-lineno">106</tt> <tt class="py-line"> </tt>
- <a name="L107"></a><tt class="py-lineno">107</tt> <tt class="py-line"> <tt class="py-name">stopwords</tt> <tt class="py-op">=</tt> <tt class="py-name">None</tt> </tt>
- <a name="L108"></a><tt class="py-lineno">108</tt> <tt class="py-line"> <tt class="py-keyword">if</tt> <tt class="py-name">stopwords_file</tt><tt class="py-op">:</tt> </tt>
- <a name="L109"></a><tt class="py-lineno">109</tt> <tt class="py-line"> <tt class="py-keyword">if</tt> <tt class="py-name">verbose</tt><tt class="py-op">:</tt> <tt class="py-keyword">print</tt> <tt class="py-string">'Processing stopwords...'</tt> </tt>
- <a name="L110"></a><tt class="py-lineno">110</tt> <tt class="py-line"> <tt class="py-keyword">with</tt> <tt class="py-name">open</tt><tt class="py-op">(</tt><tt class="py-name">stopwords_file</tt><tt class="py-op">,</tt> <tt class="py-string">'r'</tt><tt class="py-op">)</tt> <tt class="py-keyword">as</tt> <tt class="py-name">sw</tt><tt class="py-op">:</tt> </tt>
- <a name="L111"></a><tt class="py-lineno">111</tt> <tt class="py-line"> <tt class="py-name">stopwords</tt> <tt class="py-op">=</tt> <tt class="py-name">reader</tt><tt class="py-op">(</tt><tt class="py-name">sw</tt><tt class="py-op">.</tt><tt class="py-name">read</tt><tt class="py-op">(</tt><tt class="py-op">)</tt><tt class="py-op">)</tt> </tt>
- <a name="L112"></a><tt class="py-lineno">112</tt> <tt class="py-line"> <tt class="py-name">stopwords</tt> <tt class="py-op">=</tt> <tt class="py-op">[</tt><tt class="py-name">w</tt><tt class="py-op">.</tt><tt class="py-name">stem</tt> <tt class="py-keyword">for</tt> <tt class="py-name">w</tt> <tt class="py-keyword">in</tt> <tt class="py-name">map</tt><tt class="py-op">(</tt><tt class="py-name">stemmer</tt><tt class="py-op">,</tt> <tt class="py-name">stopwords</tt><tt class="py-op">)</tt><tt class="py-op">]</tt> </tt>
- <a name="L113"></a><tt class="py-lineno">113</tt> <tt class="py-line"> </tt>
- <a name="L114"></a><tt class="py-lineno">114</tt> <tt class="py-line"> <tt class="py-keyword">if</tt> <tt class="py-name">verbose</tt><tt class="py-op">:</tt> <tt class="py-keyword">print</tt> <tt class="py-string">'Building dictionary... '</tt> </tt>
- <a name="L115"></a><tt class="py-lineno">115</tt> <tt class="py-line"> <tt class="py-name">dictionary</tt> <tt class="py-op">=</tt> <tt id="link-6" class="py-name" targets="Module tagger.build_dict=tagger.build_dict-module.html,Function tagger.build_dict.build_dict()=tagger.build_dict-module.html#build_dict"><a title="tagger.build_dict
- tagger.build_dict.build_dict" class="py-name" href="#" onclick="return doclink('link-6', 'build_dict', 'link-6');">build_dict</a></tt><tt class="py-op">(</tt><tt class="py-name">corpus</tt><tt class="py-op">,</tt> <tt class="py-name">stopwords</tt><tt class="py-op">,</tt> <tt class="py-name">measure</tt><tt class="py-op">)</tt> </tt>
- <a name="L116"></a><tt class="py-lineno">116</tt> <tt class="py-line"> <tt class="py-keyword">with</tt> <tt class="py-name">open</tt><tt class="py-op">(</tt><tt class="py-name">output_file</tt><tt class="py-op">,</tt> <tt class="py-string">'wb'</tt><tt class="py-op">)</tt> <tt class="py-keyword">as</tt> <tt class="py-name">out</tt><tt class="py-op">:</tt> </tt>
- <a name="L117"></a><tt class="py-lineno">117</tt> <tt class="py-line"> <tt class="py-name">pickle</tt><tt class="py-op">.</tt><tt class="py-name">dump</tt><tt class="py-op">(</tt><tt class="py-name">dictionary</tt><tt class="py-op">,</tt> <tt class="py-name">out</tt><tt class="py-op">,</tt> <tt class="py-op">-</tt><tt class="py-number">1</tt><tt class="py-op">)</tt> </tt>
- </div><a name="L118"></a><tt class="py-lineno">118</tt> <tt class="py-line"> </tt>
- <a name="L119"></a><tt class="py-lineno">119</tt> <tt class="py-line"> </tt>
- <a name="L120"></a><tt class="py-lineno">120</tt> <tt class="py-line"><tt class="py-keyword">if</tt> <tt class="py-name">__name__</tt> <tt class="py-op">==</tt> <tt class="py-string">'__main__'</tt><tt class="py-op">:</tt> </tt>
- <a name="L121"></a><tt class="py-lineno">121</tt> <tt class="py-line"> </tt>
- <a name="L122"></a><tt class="py-lineno">122</tt> <tt class="py-line"> <tt class="py-keyword">import</tt> <tt class="py-name">getopt</tt> </tt>
- <a name="L123"></a><tt class="py-lineno">123</tt> <tt class="py-line"> <tt class="py-keyword">import</tt> <tt class="py-name">sys</tt> </tt>
- <a name="L124"></a><tt class="py-lineno">124</tt> <tt class="py-line"> </tt>
- <a name="L125"></a><tt class="py-lineno">125</tt> <tt class="py-line"> <tt class="py-keyword">try</tt><tt class="py-op">:</tt> </tt>
- <a name="L126"></a><tt class="py-lineno">126</tt> <tt class="py-line"> <tt class="py-name">options</tt> <tt class="py-op">=</tt> <tt class="py-name">getopt</tt><tt class="py-op">.</tt><tt class="py-name">getopt</tt><tt class="py-op">(</tt><tt class="py-name">sys</tt><tt class="py-op">.</tt><tt class="py-name">argv</tt><tt class="py-op">[</tt><tt class="py-number">1</tt><tt class="py-op">:</tt><tt class="py-op">]</tt><tt class="py-op">,</tt> <tt class="py-string">'o:s:'</tt><tt class="py-op">)</tt> </tt>
- <a name="L127"></a><tt class="py-lineno">127</tt> <tt class="py-line"> <tt class="py-name">output_file</tt> <tt class="py-op">=</tt> <tt class="py-name">options</tt><tt class="py-op">[</tt><tt class="py-number">0</tt><tt class="py-op">]</tt><tt class="py-op">[</tt><tt class="py-number">0</tt><tt class="py-op">]</tt><tt class="py-op">[</tt><tt class="py-number">1</tt><tt class="py-op">]</tt> </tt>
- <a name="L128"></a><tt class="py-lineno">128</tt> <tt class="py-line"> <tt class="py-name">stopwords_file</tt> <tt class="py-op">=</tt> <tt class="py-name">options</tt><tt class="py-op">[</tt><tt class="py-number">0</tt><tt class="py-op">]</tt><tt class="py-op">[</tt><tt class="py-number">1</tt><tt class="py-op">]</tt><tt class="py-op">[</tt><tt class="py-number">1</tt><tt class="py-op">]</tt> </tt>
- <a name="L129"></a><tt class="py-lineno">129</tt> <tt class="py-line"> <tt class="py-name">corpus</tt> <tt class="py-op">=</tt> <tt class="py-name">options</tt><tt class="py-op">[</tt><tt class="py-number">1</tt><tt class="py-op">]</tt> </tt>
- <a name="L130"></a><tt class="py-lineno">130</tt> <tt class="py-line"> <tt class="py-keyword">except</tt><tt class="py-op">:</tt> </tt>
- <a name="L131"></a><tt class="py-lineno">131</tt> <tt class="py-line"> <tt class="py-keyword">print</tt> <tt class="py-name">__doc__</tt> </tt>
- <a name="L132"></a><tt class="py-lineno">132</tt> <tt class="py-line"> <tt class="py-name">exit</tt><tt class="py-op">(</tt><tt class="py-number">1</tt><tt class="py-op">)</tt> </tt>
- <a name="L133"></a><tt class="py-lineno">133</tt> <tt class="py-line"> </tt>
- <a name="L134"></a><tt class="py-lineno">134</tt> <tt class="py-line"> <tt id="link-7" class="py-name" targets="Function tagger.build_dict.build_dict_from_files()=tagger.build_dict-module.html#build_dict_from_files"><a title="tagger.build_dict.build_dict_from_files" class="py-name" href="#" onclick="return doclink('link-7', 'build_dict_from_files', 'link-7');">build_dict_from_files</a></tt><tt class="py-op">(</tt><tt class="py-name">output_file</tt><tt class="py-op">,</tt> <tt class="py-name">corpus</tt><tt class="py-op">,</tt> <tt class="py-name">stopwords_file</tt><tt class="py-op">,</tt> <tt class="py-name">verbose</tt><tt class="py-op">=</tt><tt class="py-name">True</tt><tt class="py-op">)</tt> </tt>
- <a name="L135"></a><tt class="py-lineno">135</tt> <tt class="py-line"> </tt><script type="text/javascript">
- <!--
- expandto(location.href);
- // -->
- </script>
- </pre>
- <br />
- <!-- ==================== NAVIGATION BAR ==================== -->
- <table class="navbar" border="0" width="100%" cellpadding="0"
- bgcolor="#a0c0ff" cellspacing="0">
- <tr valign="middle">
- <!-- Tree link -->
- <th> <a
- href="module-tree.html">Trees</a> </th>
- <!-- Index link -->
- <th> <a
- href="identifier-index.html">Indices</a> </th>
- <!-- Help link -->
- <th> <a
- href="help.html">Help</a> </th>
- <!-- Project homepage -->
- <th class="navbar" align="right" width="100%">
- <table border="0" cellpadding="0" cellspacing="0">
- <tr><th class="navbar" align="center"
- ><a class="navbar" target="_top" href="http://github.com/apresta/tagger">tagger</a></th>
- </tr></table></th>
- </tr>
- </table>
- <table border="0" cellpadding="0" cellspacing="0" width="100%%">
- <tr>
- <td align="left" class="footer">
- Generated by Epydoc 3.0.1 on Wed Jun 8 01:57:46 2011
- </td>
- <td align="right" class="footer">
- <a target="mainFrame" href="http://epydoc.sourceforge.net"
- >http://epydoc.sourceforge.net</a>
- </td>
- </tr>
- </table>
- <script type="text/javascript">
- <!--
- // Private objects are initially displayed (because if
- // javascript is turned off then we want them to be
- // visible); but by default, we want to hide them. So hide
- // them unless we have a cookie that says to show them.
- checkCookie();
- // -->
- </script>
- </body>
- </html>