Python Forum
Can someone explain how does svr_rbf.predict(dates) work?
Thread Rating:
  • 0 Vote(s) - 0 Average
  • 1
  • 2
  • 3
  • 4
  • 5
Can someone explain how does svr_rbf.predict(dates) work?
#1
I am new to Python and cannot fully understand how this Python svr_rbf.predict function work. There is this predicted_price = predict_price(dates, prices, 29) which would return the predicted price in command prompt, does that mean if there is 28 rows of data in the csv, then using this 29 would predict the next day price? How does the training for RBF work in this code? Thanks.

import csv
import numpy as np
from sklearn.svm import SVR
import matplotlib.pyplot as plt


#plt.switch_backend('QT5Agg')  



dates = []
prices = []

def get_data(filename):
    with open(filename, 'r') as csvfile:
        csvFileReader = csv.reader(csvfile)
        next(csvFileReader) # skipping column names
        for row in csvFileReader:
            #dates.append(int(row[0].split('-')[0]))
            dates.append(float(row[0]))
            prices.append(float(row[1]))
    return

def predict_price(dates, prices, x):
    dates = np.reshape(dates,(len(dates), 1)) # converting to matrix of n X 1

    #svr_lin = SVR(kernel= 'linear', C= 1e3)
    #svr_poly = SVR(kernel= 'poly', C= 1e3, degree= 2)
    svr_rbf = SVR(kernel= 'rbf', C= 1e3, gamma= 0.1) # defining the support vector regression models
    svr_rbf.fit(dates, prices) # fitting the data points in the models
    #svr_lin.fit(dates, prices)
    #svr_poly.fit(dates, prices)


    plt.scatter(dates, prices, color= 'black', label= 'Data') # plotting the initial datapoints 
    plt.plot(dates, svr_rbf.predict(dates), color= 'red', label= 'RBF model') # plotting the line made by the RBF kernel
    #plt.plot(dates,svr_lin.predict(dates), color= 'green', label= 'Linear model') # plotting the line made by linear kernel
    #plt.plot(dates,svr_poly.predict(dates), color= 'blue', label= 'Polynomial model') # plotting the line made by polynomial kernel
    plt.xlabel('Date')
    plt.ylabel('Price')
    plt.title('Support Vector Regression')
    plt.legend()
    plt.show()

    #return svr_rbf.predict(x)[0], svr_lin.predict(x)[0], svr_poly.predict(x)[0]
    return svr_rbf.predict(x)[0]


get_data('aapl.csv') # calling get_data method by passing the csv file to it
print ("Dates- ", dates)
print ("Prices- ", prices)

predicted_price = predict_price(dates, prices, 29)  

print(predicted_price)
Reply


Messages In This Thread
Can someone explain how does svr_rbf.predict(dates) work? - by j2ee - Feb-22-2018, 06:50 PM

Possibly Related Threads…
Thread Author Replies Views Last Post
  How to use .predict() method Jack90_ 4 698 Nov-03-2023, 08:21 PM
Last Post: Larz60+
  How to predict Total Hours needed with List as Input? caninan 0 761 Jan-27-2023, 04:20 PM
Last Post: caninan
  Which network architecture should be chosen to predict disease severity? AlekseyPython 2 1,956 Aug-21-2021, 07:12 PM
Last Post: jefsummers
  How to predict output in a regression problem? Rezaafz 1 2,469 Apr-03-2021, 04:16 PM
Last Post: jefsummers
  Keras.Predict into Dataframe Finpyth 13 9,565 Aug-31-2020, 07:22 AM
Last Post: hussainmujtaba
  Analyze, predict the next step in a sequence. Antonio0608 0 1,971 Jul-23-2020, 05:53 PM
Last Post: Antonio0608
  Error When Using sklearn Predict Function firebird 0 2,023 Mar-21-2020, 04:34 PM
Last Post: firebird
  Predict Longitude and Latitude Using Python vibeandvisualize 1 2,247 Dec-27-2019, 12:10 PM
Last Post: Larz60+
  How to predict with date as input for DecisionTreeRegressor sandeep_ganga 0 1,784 Dec-12-2019, 03:29 AM
Last Post: sandeep_ganga
  Model.predict() always returning the same value of 1 for opencv nastyheatnor 0 5,483 Dec-14-2017, 08:20 AM
Last Post: nastyheatnor

Forum Jump:

User Panel Messages

Announcements
Announcement #1 8/1/2020
Announcement #2 8/2/2020
Announcement #3 8/6/2020