ytest[ytest<0] = 0

Assignment1/main.py

@@ -265,11 +265,11 @@ def main():
     errlog = model.train(xtrainprocessed, ytrainprocessed, loss_fn, loss_grad, args.epoch, args.lr)
     ytest = model.predict(xtestprocessed)
     ytest = ytest.astype('int')
-    ytest = [np.absolute(i) for i in ytest]
+    ytest[ytest<0] = 0
     print(ytest)
     output = [(i,ytest[i]) for i in range(len(ytest))]
     np.savetxt("prediction.csv",output,delimiter=',',fmt="%d",header="instance (id),count",comments='')
-	# np.savetxt("error.log",errlog,delimiter='\n',fmt="%f")
+    #np.savetxt("error.log",errlog,delimiter='\n',fmt="%f")
 
 if __name__ == '__main__':