MLP - DimensionMisMatch Error Just Before Training

Hi, I’m trying to train a MLP model to predict number of real roots of polynomials. x_train and y_train include array of arrays such as [[-204, 20, 13, 1, 0]] which are coefficients of polynomials. x_test and y_test include number of real roots of each polynomial such 1,2,5… I have polynomials up to 100 degree therefore 101 expresses that. I’m stuck in this error. I’m trying to give data to MLP properly but I couldn’t do it even though I tried so hard. Can you please help me about this issue?

Here is my code:

using Flux

function input()
	## x_train
	lines = Tuple(readlines("/home/user/Desktop/x_train.txt"))
	x_train = []

	for i in lines
		push!(x_train, convert(Array{Float32},eval(Meta.parse(i))))

	x_train = reshape(x_train, 2, :)
	## y_train
	lines = Tuple(readlines("/home/user/Desktop/y_train.txt"))
	y_train = []

	for i in lines
		push!(y_train, eval(Meta.parse(i)))
	y_train = reshape(y_train, 2, :)

	## x_test
	lines = Tuple(readlines("/home/user/Desktop/x_test.txt"))
	x_test = []

	for i in lines
		push!(x_test, convert(Array{Float32},eval(Meta.parse(i))))

	x_test = reshape(x_test, 1, :)

	## y_test
	lines = Tuple(readlines("/home/user/Desktop/y_test.txt"))
	y_test = []

	for i in lines
		push!(y_test, eval(Meta.parse(i)))
	y_test = reshape(y_test, 1, :)

	return x_train, x_test, y_train, y_test

x_train, x_test, y_train, y_test = input()

train_size = 18

function NeuralNetwork()
    return Chain(
            Dense(2, 9,relu),
            Dense(9,101), softmax

# Organizing the data in batches
X    = hcat(x_train,y_train)
Y    = vcat(ones(train_size),zeros(train_size))
Y = reshape(Y, 18, 2)
data = Flux.Data.DataLoader((X, Y'), batchsize=100,shuffle=true);
# Defining our model, optimization algorithm and loss function
m    = NeuralNetwork()
opt = Descent(0.05)
loss(x, y) = sum(Flux.Losses.binarycrossentropy(m(x), y))

ps = Flux.params(m)
epochs = 20
for i in 1:epochs
    Flux.train!(loss, ps, data, opt)

Error Message I Got:

ERROR: LoadError: DimensionMismatch("dimensions must match: a has dims (Base.OneTo(5),), b has dims (Base.OneTo(4),), mismatch at 1")

Thanks in advance!

My advice here is the same as LSTM Method Error - Time Series - #2 by ToucheSir work backwards and make sure all your shapes are correct.

1 Like

Thank you very much for your answer!