X_train_38,y_train_38 = get_data_by_digits(X_train,y_train,3,8)
X_test_38,y_test_38=get_data_by_digits(X_test,y_test,3,8)
# extract only 7 & 9 digits
X_train_79,y_train_79 = get_data_by_digits(X_train,y_train,7,9)
X_test_79,y_test_79=get_data_by_digits(X_test,y_test,7,9)
run_38 = {}
run_79 = {}
run_38["parameters"], run_38["costs"] = ANN.L_layer_model(X_train_38, y_train_38, layers_dims=[20, 7, 5, 1], learning_rate=0.009,
num_iterations=3000)
run_79["parameters"], run_79["costs"] = ANN.L_layer_model(X_train_79, y_train_79, layers_dims=[20, 7, 5, 1], learning_rate=0.009,
num_iterations=3000)
run_38["train_acc"]=ANN.Predict(X_train_38,y_train_38, run_38["parameters"])
run_38["test_acc"]=ANN.Predict(X_test_38,y_test_38, run_38["parameters"])
run_79["train_acc"]=ANN.Predict(X_train_79,y_train_79, run_79["parameters"])
run_79["test_acc"]=ANN.Predict(X_test_79,y_test_79, run_79["parameters"])
print("Run 7,9 labels test set Acc: "+str(run_79["test_acc"]) +"\n" )
print("Run 7,9 labels train set Acc: "+str(run_79["train_acc"]) +"\n" )
print("Run 3,8 labels test set Acc: "+str(run_38["test_acc"]) +" \n" )
print("Run 3,8 labels train set Acc: "+str(run_38["train_acc"]) +"\n" )
with open("cost_result.csv",mode="w") as f:
f.write("index,")
for i in range(len(run_79["costs"])):
f.write(str(i)+",")
f.write("\n")
