com.mayabot.nlp.common.matrix.EnumTransformMatrix Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mynlp Show documentation

Maya Nlp subproject :mynlp

There is a newer version: 4.0.0

/*
 * Copyright 2018 mayabot.com authors. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.mayabot.nlp.common.matrix;

import com.mayabot.nlp.resources.NlpResource;

import java.io.IOException;
import java.io.InputStream;

/**
 * 转移矩阵词典
 *
 * @param  标签的枚举类型
 * @author jimichan
 */
public class EnumTransformMatrix> {

    private TransformMatrix transformMatrix;

    public EnumTransformMatrix(NlpResource resource) throws IOException {
        transformMatrix = new TransformMatrix();

        try (InputStream in = resource.inputStream()) {
            transformMatrix.load(in);
        }
    }

    public int getFrequency(E from, E to) {
        return transformMatrix.getFrequency(from.name(), to.name());
    }

    public long getTotalFrequency(E from) {
        return transformMatrix.getTotalFrequency(from.name());
    }

    public long getTotalFrequency() {
        return transformMatrix.getTotalFrequency();
    }

    @Override
    public String toString() {
        return transformMatrix.toString();
    }

    /**
     * 获取转移概率
     *
     * @param from
     * @param to
     * @return double
     */
    public double getTP(E from, E to) {
        Double x = transformMatrix.getTP(from.name(), to.name());
        if (x != null) {
            return x.doubleValue();
        } else {
            return 0;
        }
    }

}