#vulnerable code 
                             @Override
	public int getNumberOfFeatures(){
		List<?> coefs = getCoefs();

		NDArray input = (NDArray)coefs.get(0);

		int[] shape = NDArrayUtil.getShape(input);

		return shape[0];
	} 
                            #location 5 
                            #vulnerability type NULL_DEREFERENCE
#fixed code
                             @Override
	public int getNumberOfFeatures(){
		List<? extends HasArray> coefs = getCoefs();

		return NeuralNetworkUtil.getNumberOfFeatures(coefs);
	}