Re: [Classifier4j-devel] Bayesian with multiple categories

[Nick L. <ni...@ma...>]

package net.sf.classifier4J.vector;

import java.io.Serializable;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;
import java.util.Set;


public class MyHashMapTermVectorStorage implements TermVectorStorage, Serializable {
    private static final long serialVersionUID = 1L;
    private Map storage;
    
    
    public MyHashMapTermVectorStorage(int amount)
    {
        storage = new HashMap(amount);
    }
    

    
    public MyHashMapTermVectorStorage()
    {
        storage = new HashMap();
    }
    
    /**
     * @see net.sf.classifier4J.vector.TermVectorStorage#addTermVector(java.lang.String, net.sf.classifier4J.vector.TermVector)
     */
    public void addTermVector(String category, TermVector termVector) {
        //storage.put(category, termVector);        
        //modified: Abelssoft, Sven Abels, 16.03.2005:
              
        TermVector old=(TermVector)storage.get(category);
        if (old==null) storage.put(category, termVector);
        else
        {
            old.add(termVector);
            storage.put(category, old);
        }
    }

    /**
     * @see net.sf.classifier4J.vector.TermVectorStorage#getTermVector(java.lang.String)
     */
    public TermVector getTermVector(String category) {
        return (TermVector) storage.get(category);
    }
    
    public int size()
    {
        if (storage==null) return 0;
        return storage.size();
    }

}