Skip to content

Commit cb1519e

Browse files
committed
complete recurrent.
Signed-off-by: Robert Altena <[email protected]>
1 parent 621020f commit cb1519e

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

dl4j-examples/src/main/java/org/deeplearning4j/examples/recurrent/word2vecsentiment/SentimentExampleIterator.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232

3333
import java.io.File;
3434
import java.io.IOException;
35+
import java.nio.charset.Charset;
3536
import java.util.ArrayList;
3637
import java.util.Arrays;
3738
import java.util.List;
@@ -100,13 +101,13 @@ private DataSet nextDataSet(int num) throws IOException {
100101
if(cursor % 2 == 0){
101102
//Load positive review
102103
int posReviewNumber = cursor / 2;
103-
String review = FileUtils.readFileToString(positiveFiles[posReviewNumber]);
104+
String review = FileUtils.readFileToString(positiveFiles[posReviewNumber], (Charset)null);
104105
reviews.add(review);
105106
positive[i] = true;
106107
} else {
107108
//Load negative review
108109
int negReviewNumber = cursor / 2;
109-
String review = FileUtils.readFileToString(negativeFiles[negReviewNumber]);
110+
String review = FileUtils.readFileToString(negativeFiles[negReviewNumber], (Charset)null);
110111
reviews.add(review);
111112
positive[i] = false;
112113
}
@@ -253,7 +254,7 @@ public boolean isPositiveReview(int index){
253254
* @throws IOException If file cannot be read
254255
*/
255256
public INDArray loadFeaturesFromFile(File file, int maxLength) throws IOException {
256-
String review = FileUtils.readFileToString(file);
257+
String review = FileUtils.readFileToString(file, (Charset)null);
257258
return loadFeaturesFromString(review, maxLength);
258259
}
259260

dl4j-examples/src/main/java/org/deeplearning4j/examples/recurrent/word2vecsentiment/Word2VecSentimentRNN.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
/*******************************************************************************
1+
/* *****************************************************************************
22
* Copyright (c) 2015-2019 Skymind, Inc.
33
*
44
* This program and the accompanying materials are made available under the
@@ -40,6 +40,7 @@
4040

4141
import java.io.File;
4242
import java.net.URL;
43+
import java.nio.charset.Charset;
4344

4445
/**Example: Given a movie review (raw text), classify that movie review as either positive or negative based on the words it contains.
4546
* This is done by combining Word2Vec vectors and a recurrent neural network model. Each word in a review is vectorized
@@ -118,7 +119,7 @@ public static void main(String[] args) throws Exception {
118119

119120
//After training: load a single example and generate predictions
120121
File shortNegativeReviewFile = new File(FilenameUtils.concat(DATA_PATH, "aclImdb/test/neg/12100_1.txt"));
121-
String shortNegativeReview = FileUtils.readFileToString(shortNegativeReviewFile);
122+
String shortNegativeReview = FileUtils.readFileToString(shortNegativeReviewFile, (Charset)null);
122123

123124
INDArray features = test.loadFeaturesFromString(shortNegativeReview, truncateReviewsToLength);
124125
INDArray networkOutput = net.output(features);

0 commit comments

Comments
 (0)