Skip to content

Commit 72c4115

Browse files
committed
RiFyFi_init
1 parent 8a6e111 commit 72c4115

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/RiFyFi_VDG/src/RiFyFi_VDG.jl

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,7 @@ end
289289

290290
""" Function that load data from the CSV file for Synthetic database """
291291
function loadCSV_Synthetic(Param_Data)
292+
@info "Loading Data"
292293
augmentationType = Param_Data.Augmentation_Value.augmentationType
293294
ChunkSize = Param_Data.Chunksize
294295
Param_Data.nbTx = Param_Data.nbTx
@@ -318,6 +319,7 @@ function loadCSV_Synthetic(Param_Data)
318319
end
319320
end
320321
# Labels
322+
@info "Before loading Data"
321323
fileLabelTest= "$(savepath)/bigLabelsTest.csv"
322324
Y_testTemp = Matrix(DataFrame(CSV.File(fileLabelTest;types=Int64,header=false)))
323325
fileLabelTrain= "$(savepath)/bigLabelsTrain.csv"
@@ -331,7 +333,7 @@ function loadCSV_Synthetic(Param_Data)
331333
X_test = zeros(Float32, Param_Data.Chunksize,2,nbTest)
332334
Y_train = zeros(Param_Data.nbTx,nbTrain)
333335
Y_test = zeros(Param_Data.nbTx,nbTest)
334-
336+
@info "After Loading Data"
335337
for i in 1:size(X_trainTemp)[1]
336338
X_train[:,1,i]=X_trainTemp[i,1:Param_Data.Chunksize]
337339
X_train[:,2,i]=X_trainTemp[i,Param_Data.Chunksize+1:Param_Data.Chunksize+Param_Data.Chunksize]
@@ -346,7 +348,9 @@ function loadCSV_Synthetic(Param_Data)
346348
for i in 1:size(Y_testTemp)[1]
347349
Y_test[Y_testTemp[i]+1,i]=1
348350
end
351+
@info "return Data"
349352
return (X_train,Y_train,X_test,Y_test)
353+
350354
end
351355

352356

0 commit comments

Comments
 (0)