diff --git a/src/TensorFlowNET.Core/Framework/graph_util_impl.cs b/src/TensorFlowNET.Core/Framework/graph_util_impl.cs index ad5c61eb..f9514173 100644 --- a/src/TensorFlowNET.Core/Framework/graph_util_impl.cs +++ b/src/TensorFlowNET.Core/Framework/graph_util_impl.cs @@ -78,7 +78,7 @@ namespace Tensorflow } // Gets map of variables and the associated data. - NDArray returned_variables = null; + NDArray[] returned_variables = null; if (variable_names != null) returned_variables = sess.run(variable_names); diff --git a/src/TensorFlowNET.Core/Sessions/_FetchHandler.cs b/src/TensorFlowNET.Core/Sessions/_FetchHandler.cs index 96209934..a46decb1 100644 --- a/src/TensorFlowNET.Core/Sessions/_FetchHandler.cs +++ b/src/TensorFlowNET.Core/Sessions/_FetchHandler.cs @@ -82,6 +82,9 @@ namespace Tensorflow case "String": full_values.Add(float.NaN); break; + case "Char": + full_values.Add(float.NaN); + break; default: throw new NotImplementedException($"build_results tensor_values[0] {tensor_values[0].dtype.Name}"); } diff --git a/test/TensorFlowNET.Examples/TextProcessing/NER/LstmCrfNer.cs b/test/TensorFlowNET.Examples/TextProcessing/NER/LstmCrfNer.cs index 7e324c56..146bc080 100644 --- a/test/TensorFlowNET.Examples/TextProcessing/NER/LstmCrfNer.cs +++ b/test/TensorFlowNET.Examples/TextProcessing/NER/LstmCrfNer.cs @@ -80,7 +80,7 @@ namespace TensorFlowNET.Examples.Text.NER private float run_epoch(Session sess, CoNLLDataset train, CoNLLDataset dev, int epoch) { - NDArray results = null; + NDArray[] results = null; // iterate over dataset var batches = minibatches(train, hp.batch_size); diff --git a/test/TensorFlowNET.Examples/TextProcessing/Word2Vec.cs b/test/TensorFlowNET.Examples/TextProcessing/Word2Vec.cs index 8d1c323e..51dc270d 100644 --- a/test/TensorFlowNET.Examples/TextProcessing/Word2Vec.cs +++ b/test/TensorFlowNET.Examples/TextProcessing/Word2Vec.cs @@ -97,7 +97,7 @@ namespace TensorFlowNET.Examples if (step % eval_step == 0 || step == 1) { print("Evaluation..."); - var sim = sess.run(cosine_sim_op, new FeedItem(X, x_test)); + var sim = sess.run(cosine_sim_op, new FeedItem(X, x_test))[0]; foreach(var i in range(len(eval_words))) { var nearest = (0f - sim[i]).argsort()