/* * ****************************************************************************** * * * * * * This program and the accompanying materials are made available under the * * terms of the Apache License, Version 2.0 which is available at * * https://www.apache.org/licenses/LICENSE-2.0. * * * * See the NOTICE file distributed with this work for additional * * information regarding copyright ownership. * * 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. * * * * SPDX-License-Identifier: Apache-2.0 * ***************************************************************************** */ package org.deeplearning4j; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.time.StopWatch; import org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable; import org.deeplearning4j.models.embeddings.loader.WordVectorSerializer; import org.deeplearning4j.models.word2vec.wordstore.VocabCache; import org.deeplearning4j.nn.conf.WorkspaceMode; import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Timeout; import org.junit.jupiter.api.io.TempDir; import org.nd4j.linalg.api.buffer.DataType; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.factory.Nd4j; import org.nd4j.common.io.ClassPathResource; import org.nd4j.common.primitives.Pair; import java.io.File; import java.util.ArrayList; import java.util.List; @Slf4j @Timeout(300) public class TsneTest extends BaseDL4JTest { @TempDir public File testDir; @Override public DataType getDataType() { return DataType.FLOAT; } @Override public DataType getDefaultFPDataType() { return DataType.FLOAT; } }