Tensorflow NN输入维度

我有一个形状为(8, 8, 2)的数组,希望用这个数组训练一个神经网络。我还想在训练数据中添加一个整数,像这样:[array, int],但这样会得到一个形状为(2,)的数组,无法用于训练。有什么方法可以创建一个TensorFlow可以接受的数组,或者用这个新数组来训练神经网络吗?

我尝试将整数转换为向量并添加到数组中,这样得到的形状为(9, 8, 2),但这会导致很多冗余信息。

my_array = [[(rook, b), (horse, b), (bishop, b), (queen, b), (king, b), (bishop, b), (horse, b), (rook, b)],                       [(pawn, b) for i in range(8)],                       [(0, 0) for i in range(8)],                       [(0, 0) for i in range(8)],                       [(0, 0) for i in range(8)],                       [(0, 0) for i in range(8)],                       [(pawn, w) for i in range(8)],                       [(rook, w), (horse, w), (bishop, w), (queen, w), (king, w), (bishop, w), (horse, w), (rook, w)]]my_int = 1I_tried = my_array+[[(my_int, my_int) for i in range(8)]]print(numpy.array(my_array).shape) # --> (8, 8, 2)print(numpy.array(I_tried).shape) # --> (2,)

回答:

你可以考虑使用元组而不是列表,这取决于你进行这种拼接的目的。但你可以尝试这样做:

my_int = 1second_array = [(my_int, my_int) for i in range(8)]I_tried = (my_array, second_array)

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注