Skip to content

Commit 72e96b9

Browse files
committed
text classification train works!!!!
1 parent 900d45c commit 72e96b9

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

test/TensorFlowNET.Examples/TextProcess/DataHelpers.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ public static (NDArray, NDArray, int) build_char_dataset(string step, string mod
2525
char_dict["<unk>"] = 1;
2626
foreach (char c in alphabet)
2727
char_dict[c.ToString()] = char_dict.Count;
28-
29-
var contents = File.ReadAllLines(TRAIN_PATH);
28+
var random=new Random(42);
29+
var contents = File.ReadAllLines(TRAIN_PATH).OrderBy(line=>random.NextDouble()).ToArray();
3030
var size = limit == null ? contents.Length : limit.Value;
3131

3232
var x = new NDArray(np.int32, new Shape(size, document_max_len));

test/TensorFlowNET.Examples/TextProcess/TextClassificationTrain.cs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,8 @@ protected virtual bool RunWithImportedGraph(Session sess, Graph graph)
9191
[model_y] = y_batch,
9292
[is_training] = true,
9393
};
94+
//Console.WriteLine("x: " + x_batch.ToString() + "\n");
95+
//Console.WriteLine("y: " + y_batch.ToString());
9496
// original python:
9597
//_, step, loss = sess.run([model.optimizer, model.global_step, model.loss], feed_dict = train_feed_dict)
9698
var result = sess.run(new ITensorOrOperation[] { optimizer, global_step, loss }, train_feed_dict);

0 commit comments

Comments
 (0)