Я работаю с моделью предсказания covid-19 на sklearn. Я пытался сделать это в Tensorflow, но он не работает правильно. Мне было интересно, есть ли какой-либо эквивалент этой специфической c модели sklearn в Tensorflow 2.x. Если нет эквивалента, то это ответ.
Код:
from sklearn.model_selection import RandomizedSearchCV, train_test_split
from sklearn.svm import SVR
from sklearn.metrics import mean_squared_error, mean_absolute_error
import operator
kernel = ['poly', 'sigmoid', 'rbf']
c = [0.01, 0.1, 1, 10]
gamma = [0.01, 0.1, 1]
epsilon = [0.01, 0.1, 1]
shrinking = [True, False]
svm_grid = {'kernel':kernel, 'C':c, 'gamma':gamma,'epsilon':epsilon,'shrinking':shrinking}
svm = SVR()
svm_search = RandomizedSearchCV(svm, svm_grid, scoring = 'neg_mean_squared_error', cv=3, return_train_score=True, n_jobs=-1, n_iter=100, verbose=1)
X_train_confirmed, X_test_confirmed, y_train_confirmed, y_test_confirmed = train_test_split(days, confirmed, test_size=0.15, shuffle=False)
svm_search.fit(X_train_confirmed, y_train_confirmed.ravel())
Остальной код здесь: GITHUB COVID-19 CL
In [13]:
svm_search.best_params_
Out[13]:
{'C': 0.1, 'epsilon': 1, 'gamma': 0.1, 'kernel': 'poly', 'shrinking': False}
In [14]:
svm_confirmed = svm_search.best_estimator_
svm_prediction = svm_confirmed.predict(future_forecast_days)
На любое предложение по улучшению вопроса будет дан ответ