In order to provide accurate and reliable prediction of short-term traffic flow to realise intelligent transportation control, support vector machine (SVM) regression method is established to predict short-term traffic flow. Then, parameter selection optimisation model for SVM is studied. Support vector penalty coefficient and the parameters of the kernel function play an important role in learning precision and generalisation ability of regression model. So, a kind of improved artificial fish swarm algorithm is used to optimise the SVM regression to select the optimal parameters. The experiment results show that the proposed scheme can effectively reduce mean absolute percentage error and mean square error in the real traffic flow forecasting. The proposed scheme can improve the prediction precision of the short-term traffic flow.