0

我已经使用sklow中的tensorforest以及以下模型文件构建了一个模型

  1. 检查点
  2. 图.pbtxt
  3. 模型.ckpt-1.meta
  4. model.ckpt-1-00000-of-00001
  5. events.out.tfevents.1495890133.COGCONSs-MacBook-Pro.local

谁能告诉我如何编写代码来冻结和倾斜模型?我是否必须使用占位符,就像它在 tensorflow 中所做的那样,还是 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)
4

0 回答 0