Прощупывание модели skflow

Я построил модель, используя тензорный лес в sklow вместе со следующими файлами модели

  1. контрольно-пропускной пункт
  2. graph.pbtxt
  3. model.ckpt-1.meta
  4. model.ckpt-1-00000-оф-00001
  5. events.out.tfevents.1495890133.COGCONSs-MacBook-Pro.local

Может кто-нибудь сказать мне, как написать код для заморозки и sklow модели? Нужно ли использовать заполнители, как это сделано в tenorflow или он отличается для skflow? Вот код

    y_train = pd.factorize(df_train['data.0.sleeping'])[0]
    y_test = pd.factorize(df_test['data.0.sleeping'])[0]
    df_test=df_test.iloc[:, :-1]
    print(df_test.head(3))
    hparams = tf.contrib.tensor_forest.python.tensor_forest.ForestHParams(
             num_trees=3, max_nodes=1000, num_classes=2, num_features=4)
    clf = tf.contrib.learn.TensorForestEstimator(hparams, model_dir='model/')
    clf.fit(x=df_train[features], y=y_train,steps=100)

0 ответов

Другие вопросы по тегам