相关疑难解决方法(0)

如何将Scikit-Learn-Keras模型保存到持久性文件中(pickle/hd5/json/yaml)

我有以下代码,使用Keras Scikit-Learn Wrapper:

from keras.models import Sequential
from sklearn import datasets
from keras.layers import Dense
from sklearn.model_selection import train_test_split
from keras.wrappers.scikit_learn import KerasClassifier
from sklearn.model_selection import StratifiedKFold
from sklearn.model_selection import cross_val_score
from sklearn import preprocessing
import pickle
import numpy as np
import json

def classifier(X, y):
    """
    Description of classifier
    """
    NOF_ROW, NOF_COL =  X.shape

    def create_model():
        # create model
        model = Sequential()
        model.add(Dense(12, input_dim=NOF_COL, init='uniform', activation='relu'))
        model.add(Dense(6, init='uniform', activation='relu'))
        model.add(Dense(1, init='uniform', activation='sigmoid'))
        # Compile model
        model.compile(loss='binary_crossentropy', optimizer='adam', …
Run Code Online (Sandbox Code Playgroud)

python persistence pickle scikit-learn keras

22
推荐指数
4
解决办法
8880
查看次数

标签 统计

keras ×1

persistence ×1

pickle ×1

python ×1

scikit-learn ×1