001/*
002 * (C) Copyright 2010 Nuxeo SAS (http://nuxeo.com/) and contributors.
003 *
004 * All rights reserved. This program and the accompanying materials
005 * are made available under the terms of the GNU Lesser General Public License
006 * (LGPL) version 2.1 which accompanies this distribution, and is available at
007 * http://www.gnu.org/licenses/lgpl.html
008 *
009 * This library is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012 * Lesser General Public License for more details.
013 *
014 * Contributors:
015 *     Nuxeo - initial API and implementation
016 */
017
018package org.nuxeo.ecm.platform.categorization.categorizer.tfidf;
019
020import java.io.IOException;
021
022import org.nuxeo.ecm.platform.categorization.service.Categorizer;
023import org.nuxeo.ecm.platform.categorization.service.CategorizerFactory;
024
025public class TfIdfCategorizerFactory implements CategorizerFactory {
026
027    public Categorizer loadInstance(String modelFile, boolean readonly) {
028        try {
029            TfIdfCategorizer categorizer = TfIdfCategorizer.load(modelFile);
030            if (readonly) {
031                categorizer.disableUpdate();
032            }
033            return categorizer;
034        } catch (IOException | ClassNotFoundException e) {
035            throw new RuntimeException(e);
036        }
037    }
038
039}