|
Java example source code file (WordConverter.java)
The WordConverter.java Java example source code/* * * * Copyright 2015 Skymind,Inc. * * * * 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 org.deeplearning4j.text.movingwindow; import org.deeplearning4j.models.word2vec.Word2Vec; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.factory.Nd4j; import org.nd4j.linalg.util.FeatureUtil; import java.util.ArrayList; import java.util.List; public class WordConverter { private List<String> sentences = new ArrayList<>(); private Word2Vec vec; private List<Window> windows; public WordConverter(List<String> sentences,Word2Vec vec) { this.sentences = sentences; this.vec = vec; } public static INDArray toInputMatrix(List<Window> windows,Word2Vec vec) { int columns = vec.lookupTable().layerSize() * vec.getWindow(); int rows = windows.size(); INDArray ret = Nd4j.create(rows,columns); for(int i = 0; i < rows; i++) { ret.putRow(i, WindowConverter.asExampleMatrix(windows.get(i),vec)); } return ret; } public INDArray toInputMatrix() { List<Window> windows = allWindowsForAllSentences(); return toInputMatrix(windows,vec); } public static INDArray toLabelMatrix(List<String> labels,List Other Java examples (source code examples)Here is a short list of links related to this Java WordConverter.java source code file: |
... this post is sponsored by my books ... | |
#1 New Release! |
FP Best Seller |
Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.
A percentage of advertising revenue from
pages under the /java/jwarehouse
URI on this website is
paid back to open source projects.